部分源码:

报错如下:

网上有帖子说是多进程的问题,要将 num_works 修改为0,其实个人感觉这个error不管多进程的事。

关于多进程报错 可以参见我另一篇文章:

【pytorch】RuntimeError: Integer division of tensors using div or / is no longer supported【解决】。

这里经过检查修改,我是出错在:torchvision.transforms.Totensor()  没有加括号

加上之后,错误解决。

 

 

 

更多推荐

解决:TypeError: Caught TypeError in DataLoader worker process 0.