torch.optim.Optimizer.load_state_dict

Optimizer.load_state_dict(state_dict) [source]

Загружает состояние оптимизатора.

Параметры

state_dict (словарь) – состояние оптимизатора. Должно быть объектом, возвращаемым вызовом state_dict().