torch.optim.Optimizer.load_state_dict

Optimizer.load_state_dict(state_dict)

載入優化器的狀態。

參數

state_dict (dict) – 優化器狀態。應該是呼叫 state_dict() 後傳回的物件。

注意

參數的名稱（如果它們存在於 state_dict() 中每個參數組的 “param_names” 鍵之下）不會影響載入過程。若要將參數名稱用於自訂情況（例如，當載入的 state dict 中的參數與在優化器中初始化的參數不同時），應實作一個自訂的 register_load_state_dict_pre_hook 以相應地調整載入的 dict。 如果載入的 state dict 的 param_groups 中存在 param_names，它們將被保存，並覆蓋優化器狀態中當前的名稱（如果存在）。如果載入的 state dict 中不存在它們，則優化器的 param_names 將保持不變。