pytorch基础入门教程/一小时学会pytorch

来源  https://blog.csdn.net/weixin_41070748/article/details/89890330?spm=1001.2014.3001.5502

 

运行训练模型的代码时报错   BrokenPipeError: [Errno 32] Broken pipe

解决办法: 修改调用torch.utils.data.DataLoader()函数时的 num_workers 参数,该参数是指在进行数据集加载时，启用的线程数目。截止当前2018年5月9日11:15:52，如官方未解决该BUG，则可以通过修改num_works参数为 0 ，只启用一个主进程加载数据集，避免在windows使用多线程即可。