train_refine.py运行一段时间后会卡死

运行了一段时间后卡死。显卡利用率为 100%，一直卡死。需要手动关闭进程。