在使用pytorch在对MNIST数据集进行预览时,出现了TypeError: 'module' object is not callable的错误:
上报错信息图如下:
[在这里插入图片描述...]
从图中可以看出,报错位置为第35行,也就是如下位置的错误:
images, labels = next(iter(data_loader_train))
在经过多次的检查发现,引起MNIST数据集无法显现的问题不是由于这一行所引起的...,而是由于缺少了对图片进行处理,在加载数据代码的前添加上如下的代码:
transform = transforms.Compose([
transforms.ToTensor(),...transforms.Normalize(mean=(0.5, 0.5, 0.5), std=(0.5, 0.5, 0.5))
])
此时问题就已经解决了
下面完整的代码贴出来:
1.获取手写数字的训练集和测试集...# 2.root 存放下载的数据集的路径
# 3.transform用于指定导入数据集需要对数据进行哪种操作
# 4.train是指定在数据集下完成后需要载入数据哪部分
import torch
import
云服务器
即时通信 IM
ICP备案
云直播
实时音视频
