Oh, that point was neglected, “eval” and “train” are not needed if you are not using something special like
BatchNorm
See https://pytorch.org/docs/stable/generated/torch.nn.Module.html?highlight=eval#torch.nn.Module.eval 7