- 博客(3)
- 收藏
- 关注
RSS订阅原创 为什么使用模型预测时模型会再训练一次
因为在模型里的训练代码没有封装成函数，直接全局执行，import selfResnet.py 就开始训练了。把训练部分封装成函数，这样在预测代码import这个文件时，训练部分的代码就不会运行了。
2024-01-30 17:44:18
278
1
原创 一维深度残差收缩网络报错调试
最后一个batch只有[13, 1, 1024]了，故报错。可以在定义dataloader时将drop_last参数设置为True，这样最后一个batch如果数据不足时就会被舍弃，而不会报错。另外，也可以在数据集的 len 函数中返回整除batch_size的长度来避免最后一个batch报错。将输入reshape成[batch_size, channel, 1024]即可。表示输入为一个16通道的特征，1和16不匹配，故报错。表示卷积核有64个，通道数为1，大小为7，
2023-07-08 21:16:51
514
原创 batch、epoch详细解释
假设有一组数据集由5000个样本构成，训练集3000个，测试集2000个，设置batch_size=10意味着每一步训练10个样本，因此每一个epoch会训练多少步，取决于样本个数及设置的batch_size，每一步返回的是这十个的平均损失，样本遍历完一次为一轮，也即一个epoch。每一个epoch是先在训练集训练参数，然后在测试集上测试准确率。因此每一个epoch训练结束会把整个数据集全部遍历一遍。
2023-07-08 21:16:19
412
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人