在训练过程中跑验证集导致显存out of memory问题。