Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
PaddlePaddle
PaddleHub
提交
83dd8621
P
PaddleHub
项目概览
PaddlePaddle
/
PaddleHub
大约 1 年 前同步成功
通知
282
Star
12117
Fork
2091
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
200
列表
看板
标记
里程碑
合并请求
4
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
P
PaddleHub
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
200
Issue
200
列表
看板
标记
里程碑
合并请求
4
合并请求
4
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
提交
83dd8621
编写于
4月 01, 2019
作者:
W
wuzewu
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
record name prefix to module_desc
上级
d8ed5400
变更
1
显示空白变更内容
内联
并排
Showing
1 changed file
with
12 addition
and
4 deletion
+12
-4
paddle_hub/module/module.py
paddle_hub/module/module.py
+12
-4
未找到文件。
paddle_hub/module/module.py
浏览文件 @
83dd8621
...
@@ -108,7 +108,6 @@ class Module(object):
...
@@ -108,7 +108,6 @@ class Module(object):
self
.
default_signature
=
None
self
.
default_signature
=
None
self
.
module_info
=
None
self
.
module_info
=
None
self
.
processor
=
None
self
.
processor
=
None
self
.
name
=
"temp"
# TODO(wuzewu): print more module loading info log
# TODO(wuzewu): print more module loading info log
if
name
:
if
name
:
self
.
_init_with_name
(
name
=
name
)
self
.
_init_with_name
(
name
=
name
)
...
@@ -202,14 +201,15 @@ class Module(object):
...
@@ -202,14 +201,15 @@ class Module(object):
exe
=
fluid
.
Executor
(
fluid
.
CPUPlace
())
exe
=
fluid
.
Executor
(
fluid
.
CPUPlace
())
self
.
program
,
_
,
_
=
fluid
.
io
.
load_inference_model
(
self
.
program
,
_
,
_
=
fluid
.
io
.
load_inference_model
(
self
.
helper
.
model_path
(),
executor
=
exe
)
self
.
helper
.
model_path
(),
executor
=
exe
)
self
.
_recovery_parameter
(
self
.
program
)
self
.
_recover_variable_info
(
self
.
program
)
self
.
_load_processor
()
self
.
_load_processor
()
self
.
_load_assets
()
self
.
_load_assets
()
self
.
_recover_from_desc
()
self
.
_recover_from_desc
()
self
.
_generate_sign_attr
()
self
.
_generate_sign_attr
()
self
.
_recovery_parameter
(
self
.
program
)
self
.
_recover_variable_info
(
self
.
program
)
def
_init_with_signature
(
self
,
signatures
):
def
_init_with_signature
(
self
,
signatures
):
self
.
name_prefix
=
HUB_VAR_PREFIX
%
self
.
name
self
.
_process_signatures
(
signatures
)
self
.
_process_signatures
(
signatures
)
self
.
_check_signatures
()
self
.
_check_signatures
()
self
.
_generate_desc
()
self
.
_generate_desc
()
...
@@ -331,6 +331,10 @@ class Module(object):
...
@@ -331,6 +331,10 @@ class Module(object):
self
.
summary
=
utils
.
from_flexible_data_to_pyobj
(
self
.
summary
=
utils
.
from_flexible_data_to_pyobj
(
module_info
.
map
.
data
[
'summary'
])
module_info
.
map
.
data
[
'summary'
])
# recover name prefix
self
.
name_prefix
=
utils
.
from_flexible_data_to_pyobj
(
self
.
desc
.
extra_info
.
map
.
data
[
"name_prefix"
])
def
_generate_desc
(
self
):
def
_generate_desc
(
self
):
# save fluid Parameter
# save fluid Parameter
extra_info
=
self
.
desc
.
extra_info
extra_info
=
self
.
desc
.
extra_info
...
@@ -375,6 +379,10 @@ class Module(object):
...
@@ -375,6 +379,10 @@ class Module(object):
self
.
default_signature
.
name
if
self
.
default_signature
else
None
,
self
.
default_signature
.
name
if
self
.
default_signature
else
None
,
extra_info
.
map
.
data
[
'default_signature'
])
extra_info
.
map
.
data
[
'default_signature'
])
# save name prefix
utils
.
from_pyobj_to_flexible_data
(
self
.
name_prefix
,
self
.
desc
.
extra_info
.
map
.
data
[
"name_prefix"
])
# save module info
# save module info
module_info
=
extra_info
.
map
.
data
[
'module_info'
]
module_info
=
extra_info
.
map
.
data
[
'module_info'
]
module_info
.
type
=
module_desc_pb2
.
MAP
module_info
.
type
=
module_desc_pb2
.
MAP
...
@@ -520,7 +528,7 @@ class Module(object):
...
@@ -520,7 +528,7 @@ class Module(object):
return
feed_dict
,
fetch_dict
,
program
return
feed_dict
,
fetch_dict
,
program
def
get_name_prefix
(
self
):
def
get_name_prefix
(
self
):
return
HUB_VAR_PREFIX
%
self
.
name
return
self
.
name_prefix
def
get_var_name_with_prefix
(
self
,
var_name
):
def
get_var_name_with_prefix
(
self
,
var_name
):
return
self
.
get_name_prefix
()
+
var_name
return
self
.
get_name_prefix
()
+
var_name
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录