PyTorch中的model.train()有什么作用?
它调用forward()
的nn.Module
?我以为我们在调用模型时forward
会使用方法。为什么我们需要指定train()?
回答:
model.train()
告诉模型您正在训练模型。因此,在训练和测试过程中表现不同的有效层(如辍学,batchnorm等)可以知道发生了什么,因此可以相应地表现。
更多详细信息:它设置训练模式(请参阅源代码)。您可以致电model.eval()
或model.train(mode=False)
告诉您正在测试。期望train
功能训练模型有些直观,但是并没有做到这一点。它只是设置模式。
以上是 PyTorch中的model.train()有什么作用? 的全部内容, 来源链接: utcz.com/qa/420980.html