Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
PaddlePaddle
hapi
提交
b9463b72
H
hapi
项目概览
PaddlePaddle
/
hapi
通知
11
Star
2
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
4
列表
看板
标记
里程碑
合并请求
7
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
H
hapi
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
4
Issue
4
列表
看板
标记
里程碑
合并请求
7
合并请求
7
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
体验新版 GitCode,发现更多精彩内容 >>
提交
b9463b72
编写于
4月 24, 2020
作者:
L
LielinJiang
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
refine save inference model
上级
d2c796bd
变更
1
显示空白变更内容
内联
并排
Showing
1 changed file
with
5 addition
and
1 deletion
+5
-1
hapi/model.py
hapi/model.py
+5
-1
未找到文件。
hapi/model.py
浏览文件 @
b9463b72
...
@@ -1180,6 +1180,8 @@ class Model(fluid.dygraph.Layer):
...
@@ -1180,6 +1180,8 @@ class Model(fluid.dygraph.Layer):
export_for_deployment
=
True
,
export_for_deployment
=
True
,
program_only
=
False
):
program_only
=
False
):
"""
"""
Save inference model must in static mode.
Args:
Args:
dirname(str): The directory path to save the inference model.
dirname(str): The directory path to save the inference model.
model_filename(str|None): The name of file to save the inference program
model_filename(str|None): The name of file to save the inference program
...
@@ -1201,7 +1203,9 @@ class Model(fluid.dygraph.Layer):
...
@@ -1201,7 +1203,9 @@ class Model(fluid.dygraph.Layer):
Returns:
Returns:
list: The fetch variables' name list
list: The fetch variables' name list
"""
"""
# self.prepare(self.)
assert
not
fluid
.
in_dygraph_mode
(
),
'Save inference model must in static mode!'
prog
=
self
.
_adapter
.
_progs
.
get
(
'test'
,
None
)
prog
=
self
.
_adapter
.
_progs
.
get
(
'test'
,
None
)
assert
prog
,
\
assert
prog
,
\
"Model is not ready, please call `model.prepare()` first"
"Model is not ready, please call `model.prepare()` first"
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录