【pytorch】model.eval()和model.train()

eval()

self.eval()和self.train(False)等价
eval()在测试之前加，否则有输入数据即使不训练，它也会改变权值
pytorch会自己把BatchNormalization和DropOut固定住，不会取平均，而是用训练好的值。

train()

model.train()让model变成训练模式