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
相关文章
相关标签/搜索