PyCharm 调试卡在 collecting data

Author Avatar
patrickcty 11月 03, 2021

在使用 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