PyCharm 调试卡在 collecting data
在使用 PyCharm Debug PyTorch 程序的时候,经常会卡在 collecting data。根据 stackoverflow 可以知道原因在于默认设置下对多个 worker 处理的不太好。
解决方法一
在 Preferences | Build, Execution, Deployment | Python Debugger 中勾选 Gevent compatible
Gevent 是 Python 的一个协程库,而 PyTorch 代码中设置 dataloader num_workers 大于一,因此这样改能解决问题
Gevent 相关内容参考 https://www.jianshu.com/p/73ccb425a710
解决方法二
将 num_workers 设置为 0 或 1