关于DataLoader是否shuffle在VOC2007语义分割数据集上引发的问题

问题描述:

在训练过程中,训练集和验证集实时得到的F1分数相差很大,如下图:

image.png

这个问题之前从未遇到过,后来经过不断的排查,发现是因为验证集的数据加载器中shuffle设置的为False,而训练集设置的为True。

之前遇到过训练集的DataLoader未设置shuffle为True,跑出来的结果很差。

估计这个和数据集有关,以前的数据集设置的都是False,也没有出现类似的情况。

修改后:

image.png

你可能感兴趣的:(疑难问题,pytorch,神经网络,深度学习,cnn)