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

    技术2022-07-11  88

    eval()

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

    train()

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

    Processed: 0.013, SQL: 9