Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
暴走的小小菜鸟
Paddle
提交
aea05b6e
P
Paddle
项目概览
暴走的小小菜鸟
/
Paddle
与 Fork 源项目一致
Fork自
PaddlePaddle / Paddle
通知
1
Star
0
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
P
Paddle
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
体验新版 GitCode,发现更多精彩内容 >>
提交
aea05b6e
编写于
9月 09, 2017
作者:
C
chengduoZH
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
add param mean_var_names
上级
26638e9c
变更
2
隐藏空白更改
内联
并排
Showing
2 changed file
with
9 addition
and
4 deletion
+9
-4
python/paddle/trainer/config_parser.py
python/paddle/trainer/config_parser.py
+5
-4
python/paddle/trainer_config_helpers/layers.py
python/paddle/trainer_config_helpers/layers.py
+4
-0
未找到文件。
python/paddle/trainer/config_parser.py
浏览文件 @
aea05b6e
...
@@ -2366,6 +2366,7 @@ class BatchNormLayer(LayerBase):
...
@@ -2366,6 +2366,7 @@ class BatchNormLayer(LayerBase):
inputs
,
inputs
,
bias
=
True
,
bias
=
True
,
use_global_stats
=
True
,
use_global_stats
=
True
,
mean_var_names
=
None
,
moving_average_fraction
=
0.9
,
moving_average_fraction
=
0.9
,
batch_norm_type
=
None
,
batch_norm_type
=
None
,
**
xargs
):
**
xargs
):
...
@@ -2421,11 +2422,11 @@ class BatchNormLayer(LayerBase):
...
@@ -2421,11 +2422,11 @@ class BatchNormLayer(LayerBase):
psize
=
self
.
calc_parameter_size
(
image_conf
)
psize
=
self
.
calc_parameter_size
(
image_conf
)
dims
=
[
1
,
psize
]
dims
=
[
1
,
psize
]
if
mean_var_names
is
not
None
:
assert
len
(
mean_var_names
)
==
2
self
.
inputs
[
1
].
parameter_name
=
mean_var_names
[
0
]
self
.
inputs
[
2
].
parameter_name
=
mean_var_names
[
1
]
self
.
inputs
[
1
].
parameter_name
=
self
.
inputs
[
0
].
parameter_name
.
split
(
'.'
)[
0
]
+
'.'
+
\
self
.
inputs
[
1
].
parameter_name
.
split
(
'.'
)[
1
]
self
.
inputs
[
2
].
parameter_name
=
self
.
inputs
[
0
].
parameter_name
.
split
(
'.'
)[
0
]
+
'.'
+
\
self
.
inputs
[
2
].
parameter_name
.
split
(
'.'
)[
1
]
self
.
create_input_parameter
(
0
,
psize
)
self
.
create_input_parameter
(
0
,
psize
)
self
.
create_input_parameter
(
1
,
psize
,
dims
)
self
.
create_input_parameter
(
1
,
psize
,
dims
)
self
.
create_input_parameter
(
2
,
psize
,
dims
)
self
.
create_input_parameter
(
2
,
psize
,
dims
)
...
...
python/paddle/trainer_config_helpers/layers.py
浏览文件 @
aea05b6e
...
@@ -2957,6 +2957,7 @@ def batch_norm_layer(input,
...
@@ -2957,6 +2957,7 @@ def batch_norm_layer(input,
bias_attr
=
None
,
bias_attr
=
None
,
param_attr
=
None
,
param_attr
=
None
,
layer_attr
=
None
,
layer_attr
=
None
,
mean_var_names
=
None
,
batch_norm_type
=
None
,
batch_norm_type
=
None
,
moving_average_fraction
=
0.9
,
moving_average_fraction
=
0.9
,
use_global_stats
=
None
):
use_global_stats
=
None
):
...
@@ -3014,6 +3015,8 @@ def batch_norm_layer(input,
...
@@ -3014,6 +3015,8 @@ def batch_norm_layer(input,
:type param_attr: ParameterAttribute
:type param_attr: ParameterAttribute
:param layer_attr: Extra Layer Attribute.
:param layer_attr: Extra Layer Attribute.
:type layer_attr: ExtraLayerAttribute
:type layer_attr: ExtraLayerAttribute
:param mean_var_names: [mean name, variance name]
:type mean_var_names: string list
:param use_global_stats: whether use moving mean/variance statistics
:param use_global_stats: whether use moving mean/variance statistics
during testing peroid. If None or True,
during testing peroid. If None or True,
it will use moving mean/variance statistics during
it will use moving mean/variance statistics during
...
@@ -3044,6 +3047,7 @@ def batch_norm_layer(input,
...
@@ -3044,6 +3047,7 @@ def batch_norm_layer(input,
active_type
=
act
.
name
,
active_type
=
act
.
name
,
type
=
LayerType
.
BATCH_NORM_LAYER
,
type
=
LayerType
.
BATCH_NORM_LAYER
,
batch_norm_type
=
batch_norm_type
,
batch_norm_type
=
batch_norm_type
,
mean_var_names
=
mean_var_names
,
bias
=
ParamAttr
.
to_bias
(
bias_attr
),
bias
=
ParamAttr
.
to_bias
(
bias_attr
),
moving_average_fraction
=
moving_average_fraction
,
moving_average_fraction
=
moving_average_fraction
,
use_global_stats
=
use_global_stats
,
use_global_stats
=
use_global_stats
,
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录