- 博客(4)
- 收藏
- 关注
原创 CNN经典网络-Lenet(模型搭建+训练+预测+可视化)
accuracy =(predict_y == val_label).sum().item() / val_label.size(0)#将预测值与真实值进行对比,如果相等predict_y == val_label,返回为1,否则为0;predict_y = torch.max(outputs, dim=1)[1] # 寻找输出最大的索引在哪,即这个测试集属于那个类别,dim=1,维度为1上寻找,因为0是batch,1是10分类。running_loss = 0.0 # 损失清零,进行下一次五百步的训练。
2023-07-27 14:39:22
218
1
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人