本节内容不多,却十分重要。本节介绍了如何保存、加载训练的模型,便于后续的微调、模型推理与部署。
这里简单定义了一个模型。
save__checkpint()接口,保存了网络与指定文件“model.ckpt”,最后在用load_param_into_net()检查参数是否加载成功。
《昇思25天学习打卡营第9天|09-保存与加载》
于 2024-07-09 12:35:42 首次发布
本节内容不多,却十分重要。本节介绍了如何保存、加载训练的模型,便于后续的微调、模型推理与部署。
这里简单定义了一个模型。
save__checkpint()接口,保存了网络与指定文件“model.ckpt”,最后在用load_param_into_net()检查参数是否加载成功。