PyTorch中的model.train（）有什么作用？

PyTorch中的model.train（）有什么作用？

model.train()

告诉模型您正在训练模型。因此，在训练和测试过程中表现不同的有效层（如辍学，batchnorm等）可以知道发生了什么，因此可以相应地表现。

更多详细信息：它设置训练模式（请参阅源代码）。您可以致电

model.eval()

或

model.train(mode=False)

告诉您正在测试。期望

train

功能训练模型有些直观，但是并没有做到这一点。它只是设置模式。