torch.optim.Optimizer.load_state_dict

Optimizer.load_state_dict(state_dict)

加载优化器状态。

参数

state_dict (dict) – 优化器状态。应该是 state_dict() 调用返回的对象。