首页 > TAG信息列表 > imagnet
读取mini-imagnet图片并将其转为可迭代对象训练神经网络
da=torchvision.transforms.Compose([torchvision.transforms.ToTensor()])#必须有这步否则会出现 default_collate: batch must contain tensors, numpy arrays, numbers, dicts or lists； found ＜class ‘PI train_data=torchvision.datasets.ImageFolder('train1/',transfo