Pytorch保存訓練好的模型以及參數（8）

（1）代碼中部分函數講解          Pytorch中的state_dict其實就是python中的字典對象，可以將訓練中的layer（卷積層,線性層等等）保存下來；優化器對象Optimizer也有一個state_dict,其中包含了優化器的狀態以及被使用的超參數(如lr, momentum,weight_decay等等) （2）代碼 import torch import matplotl

>>阅读原文<<