一、错误描述
AttributeError:Can't pickle local object 'get_dataset.<locals>.Dataset
二、解决办法
把涉及到numworks
的值都改为0。
opts.py文件里的参数num_workers
默认值改为0。
main.py文件里num_workers
的值改为0。
val_loader = torch.utils.data.DataLoader(
Dataset(opt, 'val'),
batch_size=1,
shuffle=False,
num_workers=0,
pin_memory=True
)
其他的自己检查一下,把涉及到num_workers
的值都改为0。