记录一些奇奇怪怪的问题
- 报错AttributeError: Can’t pickle local object ‘get_dataloader..’
使用torch.utils.data.DataLoader(dataset=train_ds,batch_size=train_bs,shuffle=True,num_workers=num_workers)读取数据集的时候出现问题,原因是在 Windows 系统下,num_workers 参数在使用 PyTorch 的 torch.utils.data.DataLoader 加载数据时通常只能设置为 0,不能设置为大于 0 的值。这是因为在 Windows 系统下,Python 的多进程 multiprocessing 模块的实现方式与 Unix/Linux 系统有所不同。
解决方案:换成linux系统跑,或者把num_workers=num_workers删掉