您的位置: 首页 > 文章 > 【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变成训练模式
