torch.optim.Optimizer.load_state_dict

Optimizer.load_state_dict(state_dict)

載入最佳化器狀態。

引數

state_dict (dict) – 最佳化器狀態。應該是由呼叫 state_dict() 返回的物件。

注意

引數的名稱（如果它們存在於 state_dict() 中每個引數組的 “param_names” 鍵下）不會影響載入過程。對於自定義情況（例如載入的狀態字典中的引數與最佳化器中初始化的引數不同時），應實現自定義的 register_load_state_dict_pre_hook 來相應地調整載入的字典，以便使用引數的名稱。如果載入的狀態字典的 param_groups 中存在 param_names，它們將被儲存並覆蓋最佳化器狀態中當前的名稱（如果存在）。如果載入的狀態字典中不存在它們，則最佳化器的 param_names 將保持不變。