Pytorch深度学习：TypeError: 'builtin_function_or_method' object is not iterable 报错解决

最新推荐文章于 2024-07-27 19:50:31 发布

进击De小火柴

最新推荐文章于 2024-07-27 19:50:31 发布

阅读量3w

点赞数 5

分类专栏：深度学习文章标签： pytorch error iter tensor

本文链接：https://blog.csdn.net/xzy5210123/article/details/81067110

版权

在用Pytorch做图像分类的时候，遇到了这个BUG，因为这段代码和网上例子一样仍报错，所以很奇怪。现将解决方案记录分享

TypeError: 'builtin_function_or_method' object is not iterable

先贴上我报错部分和相关代码


import torch
import torch.utils.data as Data
import torchvision
from torchvision import transforms, datasets

# hyper parameters
BATCH_SIZE = 8

path = "DogsVSCats"
# define the data transform 
data_transform = transforms.Compose([
    transforms.ToTensor(),  # change the pixel to [0, 1.0]
    transforms.CenterCrop(224),
    transforms.Normalize([0.485, 0.456, 0.406], [0.229, 0.224, 0.225])
])

# define the data set in dictionary form
image_data = {x:datasets.ImageFolder(root=os.path.join(path, x), transform=data_transform)
              for x in ['train', 'valid']}
# define the data loader
data_loader = {x:Data.DataLoader(dataset=image_data[x], batc