部分源码:
报错如下:
网上有帖子说是多进程的问题,要将 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.
发布评论