Флаг requires_grad=True включает запись операций для autograd, чтобы потом посчитать градиенты.
Разбор
- Параметры
nn.Parameter по умолчанию с requires_grad=True.
- Входные фичи обычно без градиента, если не атакуете/не оптимизируете вход.
x.requires_grad_(True) включает флаг inplace.- Лишний graph на инференсе тратит память — используйте
torch.no_grad().
Итог
requires_grad решает, строить ли граф для обратного прохода.