torch.optim.Optimizer.step

Optimizer.step(closure: None = None) → None [исходный код]

Optimizer.step(closure:Callable[[],float]) → float

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

Параметры:

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