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

inference_mode

Вопрос

Чем torch.inference_mode() отличается от torch.no_grad()?

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

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

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

Коротко

Оба отключают построение графа. inferencemode строже: тензоры, созданные внутри, нельзя потом использовать в вычислениях с autograd, зато накладные расходы меньше. Подходит для чистого инференса, nograd для валидации внутри обучения и других случаев.

Ответ

Оба отключают построение графа. inference_mode строже: тензоры, созданные внутри, нельзя потом использовать в вычислениях с autograd, зато накладные расходы меньше. Подходит для чистого инференса, no_grad для валидации внутри обучения и других случаев.

Разбор

Оба отключают построение графа. inference_mode строже: тензоры, созданные внутри, нельзя потом использовать в вычислениях с autograd, зато накладные расходы меньше. Подходит для чистого инференса, no_grad для валидации внутри обучения и других случаев.

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

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