红色箭头处报错，并且错误提示为numpy.ndarray没有属性float。
按道理来讲从训练集数据加载器中枚举的数据不该是numpy.ndday类型。
查看train_loader函数，发现train_loader直接读取了训练集。
更正train_loader加载为：
‘numpy.ndarray‘ object has no attribute ‘float‘ - images = Variable(images.float()).to(DEVICE)
猜你喜欢
转载自blog.csdn.net/weixin_42133481/article/details/114408146
今日推荐
周排行