From e2712025a33e7f01af7b28e12a9075d595a20f32 Mon Sep 17 00:00:00 2001 From: chenzomi Date: Sun, 28 Jun 2020 09:32:06 +0800 Subject: [PATCH] change ckpoint_file_name to ckpt_file_name --- tutorials/source_en/advanced_use/on_device_inference.md | 4 ++-- tutorials/source_zh_cn/advanced_use/on_device_inference.md | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/tutorials/source_en/advanced_use/on_device_inference.md b/tutorials/source_en/advanced_use/on_device_inference.md index 210fd4985a..b95bfd1bf0 100644 --- a/tutorials/source_en/advanced_use/on_device_inference.md +++ b/tutorials/source_en/advanced_use/on_device_inference.md @@ -87,7 +87,7 @@ To perform on-device model inference using MindSpore, perform the following step ### Generating an On-Device Model File 1. After training is complete, load the generated checkpoint file to the defined network. ```python - param_dict = load_checkpoint(ckpoint_file_name=ckpt_file_path) + param_dict = load_checkpoint(ckpt_file_name=ckpt_file_path) load_param_into_net(net, param_dict) ``` 2. Call the `export` API to export the .ms model file on the device. @@ -145,7 +145,7 @@ if __name__ == '__main__': is_ckpt_exist = os.path.exists(ckpt_file_path) if is_ckpt_exist: - param_dict = load_checkpoint(ckpoint_file_name=ckpt_file_path) + param_dict = load_checkpoint(ckpt_file_name=ckpt_file_path) load_param_into_net(net, param_dict) export(net, input_data, file_name="./lenet.ms", file_format='LITE') print("export model success.") diff --git a/tutorials/source_zh_cn/advanced_use/on_device_inference.md b/tutorials/source_zh_cn/advanced_use/on_device_inference.md index 17be02578e..fb9a6a08fa 100644 --- a/tutorials/source_zh_cn/advanced_use/on_device_inference.md +++ b/tutorials/source_zh_cn/advanced_use/on_device_inference.md @@ -86,7 +86,7 @@ MindSpore进行端侧模型推理的步骤如下。 ### 生成端侧模型文件 1. 加载训练完毕所生成的CheckPoint文件至定义好的网络中。 ```python - param_dict = load_checkpoint(ckpoint_file_name=ckpt_file_path) + param_dict = load_checkpoint(ckpt_file_name=ckpt_file_path) load_param_into_net(net, param_dict) ``` 2. 调用`export`接口，导出端侧模型文件(.ms)。 @@ -144,7 +144,7 @@ if __name__ == '__main__': is_ckpt_exist = os.path.exists(ckpt_file_path) if is_ckpt_exist: - param_dict = load_checkpoint(ckpoint_file_name=ckpt_file_path) + param_dict = load_checkpoint(ckpt_file_name=ckpt_file_path) load_param_into_net(net, param_dict) export(net, input_data, file_name="./lenet.ms", file_format='LITE') print("export model success.") -- Gitee