train_refine.py运行一段时间过后就会卡死不动

跑着跑着就卡死不动了。显卡利用率100%，一直卡死。需要手动关闭进程。