AttributeError: 'Compose' 对象没有属性 'Compose'(在 Pytorch 0.2.1 中)

gop*_*zna 2 python deep-learning pytorch

这是出现此错误的代码块:

train_transforms = transforms.Compose([transforms.RandomRotation(30),
                                       transforms.RandomResizedCrop(224),
                                       transforms.RandomHorizontalFlip(),
                                       transforms.ToTensor(),
                                       transforms.Normalize([0.5, 0.5, 0.5], 
                                                            [0.5, 0.5, 0.5])])
Run Code Online (Sandbox Code Playgroud)

我试过更新我的 torchvision 但没有运气!

小智 8

问题是你有一个被调用的变量transformsfrom torchvision import transforms它具有某种类型的组合。这会覆盖您从torchvison. 因此,当您运行上面的代码时,它会调用variable不是来自torchvision模块的转换。

建议重命名变量,或者如果您使用的是 jupyter notebook,请import transforms在使用上面的代码运行单元格之前运行单元格。