PyTorch-Dataset

pytorch里面有很多数据集

首先在百度输入 pytorch 然后打开官网 然后搜索 torchvision 

然后

然后

点开具体的数据集 然后导入的时候 输入

mnist数据集下载地址

https://blog.csdn.net/qq_23096723/article/details/80982295?utm_medium=distribute.pc_relevant.none-task-blog-BlogCommendFromMachineLearnPai2-5&depth_1-utm_source=distribute.pc_relevant.none-task-blog-BlogCommendFromMachineLearnPai2-5

然后进行数据的导入

•  此时发现报错了 因为batch必须要包含tensors numpy arrays numbers dicts或者lists
• 所以我们要重新定义transform

•  总体代码如下所示