序列化与反序列化
序列化
torch.save(obj, f)
-
主要参数:
- obj:对象
- f:输出路径
-
例如
- 保存整个模型
torch.save(net, path)
- 保存模型参数
state_dict=net.state_dict()
torch.save(state_dict, path)
- 保存整个模型
反序列化
torch.load(f, map_location=None)
- 主要参数:
- f:文件路径
- map_location:指定存放位置,cpu or gpu