Pytorch加载数据是如何区分训练集和测试集？

1、 Pytorch加载数据是如何区分训练集和测试集？

2、 数据加载的语句如下图所示

3、 可以发现训练集和测试集的数据集的加载使用的是通用的加载函数

4、 该函数的第一个参数是数据集的路径

5、 第二个参数为是否参与训练过程，不参与则为测试集数据，参与则为训练集数据

6、 因此将第二个参数设置为True时则为训练集的数据集