赛题一 baseline代码运行错误

您好!
在我运行第二届计图人工智能挑战赛赛题一的baseline代码时,我发现它会报错,具体表现为CPU overflow。
并且这个问题不会受到设置的num_cpu而变化,同向对比pytorch 的该类报错,我进行了如下尝试:
1.修改opt.n_cpu=0或者opt.n_cpu=1,结果仍然是报错
2.在epoch执行前设置if __name__=='__main__':仍然会报错

以下是相应位置代码截图,以及报错内容:

图一:dataloader 设置

图二:报错代码行数位置

图三:导致的CPU overflow

之后内存会被不断占用,然后电脑会死机

这是内存占用过大导致的,您可以尝试通过减小batchsize来减少内存占用

谢谢您!已经解决啦