list has no attribute ‘to‘ 或者list has no attribute cuda()

一般是你的数据类型是list导致
解决1;list has no attribute ‘to‘ 或者list has no attribute cuda()_第1张图片
解决2:list has no attribute ‘to‘ 或者list has no attribute cuda()_第2张图片
如果以上两个都不行,我就是这样。结果发现是你的dataset定义的时候有问题
train_dataset = torchvision.datasets.ImageFolder(
root=‘/mnt/public/home/s-wanggc/datasets/CUB_200_2011/train’,
#download=True,
#train=True,
transform=transform.Transforms(size=args.image_size,s=0.5)
)
test_dataset = torchvision.datasets.ImageFolder(
root=‘/mnt/public/home/s-wanggc/datasets/CUB_200_2011/test’,
#download=True,
#train=False,
transform=transform.Transforms(size=args.image_size,s=0.5)
)
zai

你可能感兴趣的:(Python,深度学习,pytorch,机器学习)