今年遇到一个随着训练过程炸显存的原因，竟然是计数器保存loss时，保存在gpu上，没有加item()