使用Pytorch进行读取本地的MINIST数据集并进行装载
pytorch中的torchvision.datasets中自带MINIST数据集,可直接调用模块进行获取,也可以进行自定义自己的Dataset类进行读取本地数据和初始化数据。1. 直接使用pytorch自带的MNIST进行下载:缺点: 下载速度较慢,而且如果中途下载失败一般得是重新进行执行代码进行下载:# # 训练数据和测试数据的下载# 训练数据和测试数据的下载trainDataset...
原创
2019-12-30 15:48:39 ·
14979 阅读 ·
8 评论