From the other thread:



What does nn.Module's train(True)/train(False) do


model.train(False) and model.eval() have the same effect, as the latter calls in fact the former function as seen in this line of code.