Я написал нейронную сеть в PyTorch, которая использует ResNet в качестве генератора признаков, и она настроена на всю сеть.
Моя модель состоит из Resnet и нескольких слоев, которые я добавил к ней. Мой вопрос:
Когда я вызываю model.train(), переводит ли он ResNet в режим поезда или мне следует вызывать для него поезд отдельно?
Судя по всему , если вы вызовете train() для модуля, он будет рекурсивно вызывать train()
для всех дочерних элементов. Так что model.train()
- model
быть моделью, содержащей Resnet - будет достаточно.