Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
MindSpore
docs
提交
905eebb1
D
docs
项目概览
MindSpore
/
docs
通知
4
Star
2
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看板
体验新版 GitCode,发现更多精彩内容 >>
提交
905eebb1
编写于
6月 28, 2020
作者:
M
mindspore-ci-bot
提交者:
Gitee
6月 28, 2020
浏览文件
操作
浏览文件
下载
差异文件
!287 change ckpoint_file_name to ckpt_file_name
Merge pull request !287 from chenzhongming/master
上级
33553d47
e2712025
变更
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
浏览文件 @
905eebb1
...
@@ -87,7 +87,7 @@ To perform on-device model inference using MindSpore, perform the following step
...
@@ -87,7 +87,7 @@ To perform on-device model inference using MindSpore, perform the following step
### Generating an On-Device Model File
### Generating an On-Device Model File
1.
After training is complete, load the generated checkpoint file to the defined network.
1.
After training is complete, load the generated checkpoint file to the defined network.
```
python
```
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
)
load_param_into_net
(
net
,
param_dict
)
```
```
2.
Call the
`export`
API to export the
`.ms`
model file on the device.
2.
Call the
`export`
API to export the
`.ms`
model file on the device.
...
@@ -145,7 +145,7 @@ if __name__ == '__main__':
...
@@ -145,7 +145,7 @@ if __name__ == '__main__':
is_ckpt_exist
=
os
.
path
.
exists
(
ckpt_file_path
)
is_ckpt_exist
=
os
.
path
.
exists
(
ckpt_file_path
)
if
is_ckpt_exist
:
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
)
load_param_into_net
(
net
,
param_dict
)
export
(
net
,
input_data
,
file_name
=
"./lenet.ms"
,
file_format
=
'LITE'
)
export
(
net
,
input_data
,
file_name
=
"./lenet.ms"
,
file_format
=
'LITE'
)
print
(
"export model success."
)
print
(
"export model success."
)
...
...
tutorials/source_zh_cn/advanced_use/on_device_inference.md
浏览文件 @
905eebb1
...
@@ -86,7 +86,7 @@ MindSpore进行端侧模型推理的步骤如下。
...
@@ -86,7 +86,7 @@ MindSpore进行端侧模型推理的步骤如下。
### 生成端侧模型文件
### 生成端侧模型文件
1.
加载训练完毕所生成的CheckPoint文件至定义好的网络中。
1.
加载训练完毕所生成的CheckPoint文件至定义好的网络中。
```
python
```
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
)
load_param_into_net
(
net
,
param_dict
)
```
```
2.
调用
`export`
接口,导出端侧模型文件(
`.ms`
)。
2.
调用
`export`
接口,导出端侧模型文件(
`.ms`
)。
...
@@ -144,7 +144,7 @@ if __name__ == '__main__':
...
@@ -144,7 +144,7 @@ if __name__ == '__main__':
is_ckpt_exist
=
os
.
path
.
exists
(
ckpt_file_path
)
is_ckpt_exist
=
os
.
path
.
exists
(
ckpt_file_path
)
if
is_ckpt_exist
:
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
)
load_param_into_net
(
net
,
param_dict
)
export
(
net
,
input_data
,
file_name
=
"./lenet.ms"
,
file_format
=
'LITE'
)
export
(
net
,
input_data
,
file_name
=
"./lenet.ms"
,
file_format
=
'LITE'
)
print
(
"export model success."
)
print
(
"export model success."
)
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录