Pytorch 训练和测试时记得加 model.train 和 model.eval