关于使用Pytorch时,训练集模型表现很好但测试集模型表现极差的原因

2022-10-05 09:35:20

出现这一现象的原因主要有三个:

  1. 训练数据集过小,导致系统泛化能力不足
  2. 训练和测试时的model.train(), model.eval()没有设置好
  3. 训练时数据集一定要打乱,Dataloader的shuffle一定要设置为True,这是初学者很容易犯的错误,不打乱,模型很容易学到一些和实验无关的信息,比如数据的顺序等,导致过拟合。
  • 作者:无涯阁主
  • 原文链接:https://blog.csdn.net/qq_43222655/article/details/106329560
    更新时间:2022-10-05 09:35:20