土堆学习笔记——P28完整的模型训练套路（二）-Toy模板网

这篇具有很好参考价值的文章主要介绍了土堆学习笔记——P28完整的模型训练套路（二）。希望对大家有所帮助。如果存在错误或未考虑完全的地方，请大家不吝赐教，您也可以点击"举报违法"按钮提交疑问。

怎么知道模型有没有训练好？有没有达到想要的需求？
解决方法：在每轮训练之后加一个测试，在测试数据集上看效果（看损失之类的）。
注意：在测试时候不调优，仅为看效果。

下面的都在某一轮训练里
例如，他们都在for i in range(epoch):里面，具体看上一节代码
#测试步骤开始
total_test_loss = 0
with torch.no_grad():#取消梯度，不调优
	for data in tes_dataloader:
		img, targets = data
		outputs = tudui(imgs)
		loss = loss_fn(outputs, targets)
		#截止到现在，loss只是一部分数据也就是data在网络模型上的损失
		total_test_loss = total_test_loss+loss
		#accuracy看后面准确率那里的讲解
		accuracy = (outputs.argmax(1)==targets).sum()
		#targets在上面img, targets = data
		total_accuracy = total_accuracy + accuracy
print("整体数据集上的Loss：{}".format(total_test_loss))
print("整体数据集上的正确率：{}".format(total_accuracy/total_test_size))
torch.save(tudui,"tudui_{}.pth".format(i))