加载data.dataloader.DataLoader AttributeError: Can’t pickle local object ‘XXXXX’
解决方法:
将 num_workers的值改成num_workers = 0
dataloader = torch.utils.data.DataLoader(train_dataset, num_workers=0)
加载data.dataloader.DataLoader AttributeError: Can’t pickle local object ‘XXXXX’
解决方法:
将 num_workers的值改成num_workers = 0
dataloader = torch.utils.data.DataLoader(train_dataset, num_workers=0)