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

回到顶部