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

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

>>阅读原文<<