出现错误的原因是因为Dataloader的多线程问题
Google结果显示是Pytorch在win10中的debug,暂不支持多线程,把num_worker改为0问题解决; 查看官方文档结果如下: 也是建议我们把num_worker设为0,再逐步Debug检测Code。
另外,针对程序在运行模式下,num_worker>0开启多线程进行加速,程序无法运行的问题,将程序的入口,使用if __name__ == '__main__':包括起来即可。
if __name__ == '__main__':