Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
机器未来
Paddle
提交
f13d725a
P
Paddle
项目概览
机器未来
/
Paddle
与 Fork 源项目一致
Fork自
PaddlePaddle / Paddle
通知
1
Star
1
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
1
列表
看板
标记
里程碑
合并请求
0
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
P
Paddle
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
1
Issue
1
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
体验新版 GitCode,发现更多精彩内容 >>
提交
f13d725a
编写于
12月 04, 2017
作者:
T
tensor-tang
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
add mkldnn_lrn python interface and add it to simple net
上级
343b1a96
变更
2
隐藏空白更改
内联
并排
Showing
2 changed file
with
8 addition
and
0 deletion
+8
-0
paddle/gserver/tests/mkldnn_simple_net.conf
paddle/gserver/tests/mkldnn_simple_net.conf
+2
-0
python/paddle/trainer/config_parser.py
python/paddle/trainer/config_parser.py
+6
-0
未找到文件。
paddle/gserver/tests/mkldnn_simple_net.conf
浏览文件 @
f13d725a
...
...
@@ -51,6 +51,8 @@ tmp = img_pool_layer(input=tmp,
padding
=
1
,
pool_type
=
MaxPooling
())
tmp
=
img_cmrnorm_layer
(
input
=
tmp
,
size
=
5
,
scale
=
0
.
0001
,
power
=
0
.
75
)
tmp
=
fc_layer
(
input
=
tmp
,
size
=
channels
,
bias_attr
=
False
,
...
...
python/paddle/trainer/config_parser.py
浏览文件 @
f13d725a
...
...
@@ -2287,11 +2287,17 @@ class Conv3DLayer(Conv3DLayerBase):
class
NormLayer
(
LayerBase
):
def
__init__
(
self
,
name
,
inputs
,
**
xargs
):
super
(
NormLayer
,
self
).
__init__
(
name
,
'norm'
,
0
,
inputs
=
inputs
,
**
xargs
)
use_mkldnn
=
bool
(
int
(
g_command_config_args
.
get
(
"use_mkldnn"
,
0
)))
use_mkldnn
=
True
if
use_mkldnn
and
self
.
inputs
[
0
].
norm
.
norm_type
==
'cmrnorm-projection'
else
False
self
.
config
.
type
=
'mkldnn_lrn'
if
use_mkldnn
else
self
.
config
.
type
for
input_index
in
xrange
(
len
(
self
.
inputs
)):
input_layer
=
self
.
get_input_layer
(
input_index
)
norm_conf
=
self
.
config
.
inputs
[
input_index
].
norm_conf
parse_norm
(
self
.
inputs
[
input_index
].
norm
,
input_layer
.
name
,
norm_conf
)
norm_conf
.
scale
=
self
.
inputs
[
input_index
].
norm
.
scale
if
use_mkldnn
else
norm_conf
.
scale
self
.
set_cnn_layer
(
name
,
norm_conf
.
output_y
,
norm_conf
.
output_x
,
norm_conf
.
channels
,
False
)
if
norm_conf
.
norm_type
==
"cross-channel-norm"
:
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录