model.train()告诉模型您正在训练模型。因此,在训练和测试过程中表现不同的有效层(如辍学,batchnorm等)可以知道发生了什么,因此可以相应地表现。
更多详细信息:它设置训练模式(请参阅源代码)。您可以致电
model.eval()或
model.train(mode=False)告诉您正在测试。期望
train功能训练模型有些直观,但是并没有做到这一点。它只是设置模式。

model.train()告诉模型您正在训练模型。因此,在训练和测试过程中表现不同的有效层(如辍学,batchnorm等)可以知道发生了什么,因此可以相应地表现。
更多详细信息:它设置训练模式(请参阅源代码)。您可以致电
model.eval()或
model.train(mode=False)告诉您正在测试。期望
train功能训练模型有些直观,但是并没有做到这一点。它只是设置模式。