torch.optim.Optimizer.step

Optimizer.step(closure) [source]

Выполняет один шаг оптимизации (обновление параметров).

Параметры:

closure (Callable) – Замыкание, которое повторно оценивает модель и возвращает потерю. Необязательно для большинства оптимизаторов.

Примечание

Если не указано иное, эта функция не должна изменять поле .grad параметров.