推荐文章:模型保存和加载那些事(点击此处查看)
这篇文章是磐创AI技术团队的PyTorch专栏里的一篇文章,对于pytorch中的模型保存和加载的几种形式做了非常详细的讲解,主要包括以下6部分内容:
-
什么是状态字典:state_dict?
-
保存和加载推理模型
-
保存和加载 Checkpoint 用于推理/继续训练
-
在一个文件中保存多个模型
-
使用在不同模型参数下的热启动模式
-
通过设备保存/加载模型
推荐文章:模型保存和加载那些事(点击此处查看)
这篇文章是磐创AI技术团队的PyTorch专栏里的一篇文章,对于pytorch中的模型保存和加载的几种形式做了非常详细的讲解,主要包括以下6部分内容:
什么是状态字典:state_dict?
保存和加载推理模型
保存和加载 Checkpoint 用于推理/继续训练
在一个文件中保存多个模型
使用在不同模型参数下的热启动模式
通过设备保存/加载模型