【机器学习】如何在训练过程中挑选比较好的模型保存(pytorch)常用思路代码解析
先设置大的epoch,跑完一遍,绘制每个epoch的训练集和测试集的损失折线图,然后根据测试集的损失,选一个中间的epoch即可,测试集的损失,一定程度上反映了模型的预测性能根据1的结果,在训练结束后保存(常用)每个epoch保存一个模型,根据打印的最好的结果选。根据测试集(验证集)合的实验结果保存模型(可以是精度达到多少,可以是loss降低到多少,可以是记录两次损失之差,小于某个值就break)......