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

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

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

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

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

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

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