Pytorch加載模型的問題及解決方法

1. 簡單的模型加載 通常來講，保存模型是把參數所有用model.cpu().state_dict(), 而後加載模型時通常用 model.load_state_dict(torch.load(model_path))。 值得注意的是：torch.load 返回的是一個 OrderedDict.python import torch import torch.nn as nn class Net

>>阅读原文<<