torch.optim.Optimizer.load_state_dict

Optimizer.load_state_dict(state_dict) [source]

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

Параметры:

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