在用pytorch做图像分类时,训练模型报错TypeError: __init__() takes 1 positional argument but 2 were
data_transform = {
"train": transforms.Compose([transforms.RandomResizedCrop(244),
transforms.RandomHorizontalFlip(),
transforms.ToTensor(),
transforms.Normalize((0.5, 0.5, 0.5),(0.5, 0.5, 0.5))]),
"val": transforms.Compose([transforms.Resize((224, 224)),
transforms.ToTensor(),
transforms.Normalize((0.5, 0.5, 0.5),(0.5, 0.5, 0.5))]),
}
仔细检查后transforms.ToTensor()中的()漏输。