Pytorch中的model.train()和model.eval()——虽小但不能忽略_大梦冲冲冲的博客

背景【可以忽略】

最近，我花了一个月的时间在解决这样一个问题——“在代码的Train部分，每个一个epoch结束，我会使用第三方的数据集【例如：SOD数据集】进行test。而在代码的Test部分，我加载Train得到的模型权重，然后在对同一个第三方数据集进行test。最近，我花了一个月的时间在解决这样一个问题——“在代码的T