Почему в этом официальном учебнике по Pytorch нет метода .train ()?

Question

Итак, я только что выучил Pytorch, и они говорят, что вы должны перевести NN в режим поезда методом .train (), а затем при выводе в режим .eval (). Я читаю этот урок, и там нет .train () вообще. Почему это так?

https://pytorch.org/tutorials/beginner/blitz/cifar10_tutorial.html#sphx-glr-beginner-blitz-cifar10-tutorial-py

Answer 1

.train() устанавливает свойство self.training модуля равным True.Как видно из источника для nn.Module, это свойство изначально установлено на True.Таким образом, если вы не позвонили eval() до начала тренировки, вам не нужно звонить train().Но, вероятно, в любом случае это хорошая практика.

Кроме того, .train() и eval() влияют только на некоторые модули (например, dropout и batchnorm).Поэтому вам не нужно вызывать их, если вы не используете эти модули, но, опять же, это, вероятно, хорошая практика.