Карточка Junior PyTorch на практике обновлено

nn.Module

Вопрос

Что даёт наследование от nn.Module?

Попробуйте сами

Ответьте про себя или вслух, потом откройте ответ и сверьтесь.

Показать ответ и разбор

Коротко

Регистрацию параметров и подмодулей (model.parameters(), statedict), перенос на устройство через .to(), режимы train() и eval(), хуки. Слои создают в init, вычисления пишут в forward, вызывают модель как model(x), а не model.forward(x).

Ответ

Регистрацию параметров и подмодулей (model.parameters(), state_dict), перенос на устройство через .to(), режимы train() и eval(), хуки. Слои создают в init, вычисления пишут в forward, вызывают модель как model(x), а не model.forward(x).

Разбор

Регистрацию параметров и подмодулей (model.parameters(), state_dict), перенос на устройство через .to(), режимы train() и eval(), хуки. Слои создают в init, вычисления пишут в forward, вызывают модель как model(x), а не model.forward(x).

Ответьте сами, а не прочитайте

В тренажёре вопрос попадёт в повторения и вернётся, когда начнёт забываться. Плюс билеты с зачётом: 20 вопросов, не больше 2 ошибок.