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

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

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

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

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

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

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