pytorch 狀態字典:state_dict

pytorch 中的 state_dict 是一個簡單的python的字典對象,將每一層與它的對應參數創建映射關係.(如model的每一層的weights及偏置等等)html (注意,只有那些參數能夠訓練的layer纔會被保存到模型的state_dict中,如卷積層,線性層等等)python 優化器對象Optimizer也有一個state_dict,它包含了優化器的狀態以及被使用的超參數(如lr,

>>阅读原文<<