Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
PaddlePaddle
PaddleHub
提交
2c045b9c
P
PaddleHub
项目概览
PaddlePaddle
/
PaddleHub
大约 1 年 前同步成功
通知
280
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看板
体验新版 GitCode,发现更多精彩内容 >>
提交
2c045b9c
编写于
3月 11, 2019
作者:
W
wuzewu
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
add config process
上级
1d4afb60
变更
2
显示空白变更内容
内联
并排
Showing
2 changed file
with
28 addition
and
11 deletion
+28
-11
paddle_hub/commands/run.py
paddle_hub/commands/run.py
+1
-1
paddle_hub/module/module.py
paddle_hub/module/module.py
+27
-10
未找到文件。
paddle_hub/commands/run.py
浏览文件 @
2c045b9c
...
@@ -96,7 +96,7 @@ class RunCommand(BaseCommand):
...
@@ -96,7 +96,7 @@ class RunCommand(BaseCommand):
module
(
module
(
sign_name
=
self
.
args
.
signature
,
sign_name
=
self
.
args
.
signature
,
data
=
input_data
,
data
=
input_data
,
config
=
yaml_config
[
'config'
])
**
yaml_config
[
'config'
])
command
=
RunCommand
.
instance
()
command
=
RunCommand
.
instance
()
paddle_hub/module/module.py
浏览文件 @
2c045b9c
...
@@ -330,11 +330,11 @@ class Module:
...
@@ -330,11 +330,11 @@ class Module:
utils
.
from_pyobj_to_flexible_data
(
self
.
summary
,
utils
.
from_pyobj_to_flexible_data
(
self
.
summary
,
module_info
.
map
.
data
[
'summary'
])
module_info
.
map
.
data
[
'summary'
])
def
__call__
(
self
,
sign_name
,
data
,
config
=
None
):
def
__call__
(
self
,
sign_name
,
data
,
**
kwargs
):
feed_dict
,
fetch_dict
,
program
=
self
.
context
(
sign_name
)
feed_dict
,
fetch_dict
,
program
=
self
.
context
(
sign_name
,
for_test
=
True
)
#TODO(wuzewu): more option
#TODO(wuzewu): more option
program
=
program
.
clone
(
for_test
=
True
)
reader
=
self
.
processor
.
reader
(
reader
=
self
.
processor
.
reader
(
sign_name
=
sign_name
,
data_dict
=
data
)
sign_name
=
sign_name
,
data_dict
=
data
,
**
kwargs
)
feed_name_list
=
list
(
feed_name_list
=
list
(
set
([
value
.
name
for
key
,
value
in
feed_dict
.
items
()]))
set
([
value
.
name
for
key
,
value
in
feed_dict
.
items
()]))
fetch_list
=
list
(
set
([
value
for
key
,
value
in
fetch_dict
.
items
()]))
fetch_list
=
list
(
set
([
value
for
key
,
value
in
fetch_dict
.
items
()]))
...
@@ -347,20 +347,37 @@ class Module:
...
@@ -347,20 +347,37 @@ class Module:
feed
=
feeder
.
feed
(
batch
),
feed
=
feeder
.
feed
(
batch
),
fetch_list
=
fetch_list
,
fetch_list
=
fetch_list
,
return_numpy
=
False
)
return_numpy
=
False
)
self
.
processor
.
postprocess
(
sign_name
,
data_out
,
config
)
self
.
processor
.
postprocess
(
sign_name
,
data_out
,
**
kwargs
)
def
context
(
self
,
sign_name
,
trainable
=
False
):
def
context
(
self
,
sign_name
,
for_test
=
False
,
trainable
=
False
,
regularizer
=
None
,
learning_rate
=
1e-3
):
assert
sign_name
in
self
.
signatures
,
"module did not have a signature with name %s"
%
sign_name
assert
sign_name
in
self
.
signatures
,
"module did not have a signature with name %s"
%
sign_name
signature
=
self
.
signatures
[
sign_name
]
signature
=
self
.
signatures
[
sign_name
]
program
=
self
.
program
.
clone
()
program
=
self
.
program
.
clone
(
for_test
=
for_test
)
paddle_helper
.
remove_feed_fetch_op
(
program
)
paddle_helper
.
remove_feed_fetch_op
(
program
)
if
not
for_test
:
if
trainable
!=
"Default"
:
paddle_helper
.
set_parameter_trainable
(
program
,
trainable
)
paddle_helper
.
set_parameter_trainable
(
program
,
trainable
)
paddle_helper
.
set_op_attr
(
program
,
is_test
=
False
)
if
learning_rate
!=
"Default"
:
paddle_helper
.
set_parameter_learning_rate
(
program
,
learning_rate
)
if
regularizer
!=
"Default"
:
paddle_helper
.
set_parameter_regularizer
(
program
,
regularizer
)
self
.
_recovery_parameter
(
program
)
self
.
_recovery_parameter
(
program
)
self
.
_recover_variable_info
(
program
)
self
.
_recover_variable_info
(
program
)
paddle_helper
.
set_op_attr
(
program
,
is_test
=
for_test
)
#TODO(wuzewu): return feed_list and fetch_list directly
#TODO(wuzewu): return feed_list and fetch_list directly
feed_dict
=
{}
feed_dict
=
{}
fetch_dict
=
{}
fetch_dict
=
{}
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录