加载pytorch DataLoader生成的数据集时出现AttributeError: Can‘t pickle local object ‘create_torchvision_dataset_w
报错如图：解决方法：将 num_workers = 2, 改成num_workers = 0dataloader = torch.utils.data.DataLoader(train_dataset, batch_size=1, shuffle=True, num_workers=2, pin_memory=gpu)
·
报错如图：
解决方法：
将 num_workers = 2, 改成num_workers = 0
dataloader = torch.utils.data.DataLoader(
train_dataset, batch_size=1, shuffle=True, num_workers=2, pin_memory=gpu
)
汇聚全球AI编程工具，助力开发者即刻编程。
更多推荐
4
0- 0
已为社区贡献4条内容
所有评论(0)