加载pytorch DataLoader生成的数据集时出现AttributeError: Can‘t pickle local object ‘create_torchvision_dataset_w

报错如图:

加载pytorch DataLoader生成的数据集时出现AttributeError: Can‘t pickle local object ‘create_torchvision_dataset_w_第1张图片

解决方法:

将 num_workers = 2, 改成num_workers = 0


    dataloader = torch.utils.data.DataLoader(
        train_dataset, batch_size=1, shuffle=True, num_workers=2, pin_memory=gpu
    )

 

你可能感兴趣的:(pytorch,解决问题,深度学习,python,pytorch)