- 博客(3)
- 收藏
- 关注
原创 项目1 新冠预测
开始训练,首先将模型放到device上,设置训练的时候需要的数组等(如:训练的loss,验证的loss,验证时的预测结果,验证时记录最小的loss),设定循环的轮数,每轮分为中每次取出一个小bath。步骤是先筛选行的下标,筛选行的时候,训练集每隔5个取四个,剩下的一个给验证集。整体代码和训练模式相同,其区别有两点:1)模型不再计算梯度,以免占用内存,同时也不需要再将梯度清零,同时也不需要再反向传播,不需要更新参数 2)要计算每个轮次的loss,和最小的loss比较,选最小的loss保存这个模型。
2024-02-07 22:24:22 350 1
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人