Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
MindSpore
docs
提交
e2712025
D
docs
项目概览
MindSpore
/
docs
通知
5
Star
3
Fork
2
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
D
docs
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
提交
e2712025
编写于
6月 28, 2020
作者:
C
chenzomi
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
change ckpoint_file_name to ckpt_file_name
上级
ba1bcc9c
变更
2
隐藏空白更改
内联
并排
Showing
2 changed file
with
4 addition
and
4 deletion
+4
-4
tutorials/source_en/advanced_use/on_device_inference.md
tutorials/source_en/advanced_use/on_device_inference.md
+2
-2
tutorials/source_zh_cn/advanced_use/on_device_inference.md
tutorials/source_zh_cn/advanced_use/on_device_inference.md
+2
-2
未找到文件。
tutorials/source_en/advanced_use/on_device_inference.md
浏览文件 @
e2712025
...
...
@@ -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
(
ckp
oin
t_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
(
ckp
oin
t_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."
)
...
...
tutorials/source_zh_cn/advanced_use/on_device_inference.md
浏览文件 @
e2712025
...
...
@@ -86,7 +86,7 @@ MindSpore进行端侧模型推理的步骤如下。
### 生成端侧模型文件
1.
加载训练完毕所生成的CheckPoint文件至定义好的网络中。
```
python
param_dict
=
load_checkpoint
(
ckp
oin
t_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
(
ckp
oin
t_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."
)
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录