模型训练心得
文章平均质量分 67
Wmpreturn
这个作者很懒,什么都没留下…
展开
-
CNN中的conv
conv1是指神经网络中的第一个卷积层,它通常用于从输入数据中提取初步的特征。通过定义适当的卷积核大小、步长和填充,conv1层可以有效地处理输入数据,并生成有用的特征图。原创 2024-09-13 22:01:30 · 210 阅读 · 0 评论 -
创建数组为什么要使用np.array?直接创建数组不行吗?
虽然使用 Python 列表也可以进行基本的操作,但对于科学计算和大规模数据处理来说,NumPy 数组是更好的选择。创建数组时使用np.array相较于使用 Python 的内置列表或者其他数据结构有几个明显的优势。NumPy (numpy) 是一个强大的科学计算库,专门设计用于处理大型多维数组和矩阵,提供了高效的数学函数来操作这些数组。以下是为什么通常推荐使用np.array。原创 2024-09-12 10:16:55 · 334 阅读 · 0 评论 -
train_loader = DataLoader(train_dataset, batch_size=32, shuffle=True)
DataLoader是 PyTorch 提供的一个工具类,用于高效地加载和处理数据集。它可以帮助你在训练模型时更有效地管理和批量加载数据。让我们详细解析一下DataLoader。原创 2024-09-11 22:01:32 · 254 阅读 · 0 评论 -
模型训练中epoch的作用
Epoch:在训练过程中,当模型已经遍历了整个训练数据集一次,我们就说完成了一个 epoch。在这个过程中,模型会对每个样本进行前向传播(计算预测值),然后根据损失函数计算损失,并通过反向传播更新模型的参数。原创 2024-09-11 21:58:36 · 953 阅读 · 0 评论