报错如图:
解决方法:
将 num_workers = 2, 改成num_workers = 0
dataloader = 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
)