RuntimeError: DataLoader worker (pid(s) 9528, 8320) exited unexpectedly问题解决记录

仅供学习记录,若有侵权可联系删除。 方法一: num_workers设置为0,默认值为0表示不启用多进程。该方法在处理大量数据的时候还是不行,肯定要多进程。参见方法二。 方法二:将涉及多线程的代码放到if name == 'main’的范围内 if name == ‘main’: for epoch in range(3): for step, (batch_x, batch_y) in enum
相关文章
相关标签/搜索