PyTorch 1.0 中文文档：序列化的相关语义

译者：yuange250

最佳方案

保存模型的推荐方法

Pytorch主要有两种方法可用于序列化和保存一个模型。

第一种只存取模型的参数（更为推荐）：
保存参数：

torch.save(the_model.state_dict(), PATH)

读取参数：

the_model = TheModelClass(*args, **kwargs)
the_model.load_state_dict(torch.load(PATH))

第二种方法则将整个模型都保存下来：

torch.save(the_model, PATH)

读取的时候也是读取整个模型：

the_model = torch.load(PATH)

阅读全文／改进本文

PyTorch 1.0 中文文档：序列化的相关语义相关推荐

PyTorch 1.0 中文文档正式接受校对 | ApacheCN
参与方式:https://github.com/apachecn/pytorch-doc-zh/blob/master/CONTRIBUTING.md 整体进度:https://github.com/ ...
PyTorch 1.0 中文文档：常见问题解答
译者:冯宝宝我的模型报告"cuda runtime error(2): out of memory" 正如错误消息所示,您的GPU显存已耗尽.由于经常在PyTorch中处理大量数 ...
PyTorch 1.0 中文文档：Torch 脚本
译者:keyianpai 创建 Torch 脚本代码将追踪和脚本化结合起来 Torch 脚本语言参考类型表达式语句变量解析 python值的使用调试内置函数 Torch脚本是一种从PyT ...
PyTorch 1.0 中文文档：torch.onnx
译者:guobaoyo 示例:从Pytorch到Caffe2的端对端AlexNet模型这里是一个简单的脚本程序,它将一个在 torchvision 中已经定义的预训练 AlexNet 模型导出到 O ...
PyTorch 1.0 中文文档：torch.utils.model_zoo
译者:BXuan694 torch.utils.model_zoo.load_url(url, model_dir=None, map_location=None, progress=True) 由给 ...
PyTorch 1.0 中文文档：torchvision.utils
译者:BXuan694 torchvision.utils.make_grid(tensor, nrow=8, padding=2, normalize=False, range=None, scal ...
PyTorch 1.0 中文文档：torch.hub
译者:kunwuz torch.hub.load(github, model, force_reload=False, *args, **kwargs) 从github上加载一个带有预训练权重的模型. ...
PyTorch 1.0 中文文档：torch.utils.bottleneck
译者: belonHan torch.utils.bottleneck是调试瓶颈bottleneck时首先用到的工具.它总结了python分析工具与PyTorch自动梯度分析工具在脚本运行中情况. ...
PyTorch 1.0 中文文档：torch.autograd
译者:gfjiangly torch.autograd 提供类和函数,实现任意标量值函数的自动微分. 它要求对已有代码的最小改变-你仅需要用requires_grad=True关键字为需要计算梯度的声 ...

PyTorch 1.0 中文文档：序列化的相关语义

最佳方案

保存模型的推荐方法

PyTorch 1.0 中文文档：序列化的相关语义相关推荐

最新文章

热门文章