您的位置: 首页 > 文章 > 【pytorch】model.eval()和model.train() 【pytorch】model.eval()和model.train() 分类: 文章 • 2024-01-25 21:04:04 eval() self.eval()和self.train(False)等价 eval()在测试之前加,否则有输入数据即使不训练,它也会改变权值 pytorch会自己把BatchNormalization和DropOut固定住,不会取平均,而是用训练好的值。 train() model.train()让model变成训练模式