部分源码:

报错如下:

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

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

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

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

加上之后,错误解决。

 

 

 

Logo

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐