l-i-p-f / cbir_resnet

CBIR - DeepLearning - ResNet - Pytorch

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

好像有一个问题?请帮忙看一下

leonlee723 opened this issue · comments

在Cbir.ipynb里,
def dataLoader( istrain=True):
trans = transforms.Compose(
[transforms.ToTensor(),
transforms.Normalize(mean, std)]
)
data_set = datasets.CIFAR100('./', train=istrain, transform=trans)
return data_set

dataset = {}
for phase in ['train', 'val']:
dataset[phase] = dataLoader(phase=='train')

会报错,大概就是说dataLoader方法没有phase这个参数,运行不过去