提交 e9d839db 编写于 作者: T Tao Luo 提交者: GitHub

Merge pull request #1693 from wangyang59/cudnnDeconv

fixed a bug in conv_projection issue #1681
...@@ -7,6 +7,7 @@ build/ ...@@ -7,6 +7,7 @@ build/
.project .project
.cproject .cproject
.pydevproject .pydevproject
.settings/
Makefile Makefile
.test_env/ .test_env/
third_party/ third_party/
......
...@@ -724,7 +724,8 @@ class ConvProjection(ConvBaseProjection): ...@@ -724,7 +724,8 @@ class ConvProjection(ConvBaseProjection):
num_filters=None, num_filters=None,
conv_conf=None, conv_conf=None,
**xargs): **xargs):
super(ConvProjection, self).__init__(input_layer_name, **xargs) super(ConvProjection, self).__init__(input_layer_name, num_filters,
conv_conf, **xargs)
parse_conv(conv_conf, self.input_layer_name, self.proj_conf.conv_conf, parse_conv(conv_conf, self.input_layer_name, self.proj_conf.conv_conf,
num_filters) num_filters)
...@@ -742,7 +743,8 @@ class ConvTransProjection(ConvBaseProjection): ...@@ -742,7 +743,8 @@ class ConvTransProjection(ConvBaseProjection):
num_filters=None, num_filters=None,
conv_conf=None, conv_conf=None,
**xargs): **xargs):
super(ConvTransProjection, self).__init__(input_layer_name, **xargs) super(ConvTransProjection, self).__init__(input_layer_name, num_filters,
conv_conf, **xargs)
parse_conv( parse_conv(
conv_conf, conv_conf,
......
...@@ -156,6 +156,7 @@ layers { ...@@ -156,6 +156,7 @@ layers {
} }
inputs { inputs {
input_layer_name: "img" input_layer_name: "img"
input_parameter_name: "___mixed_6__.w1"
proj_conf { proj_conf {
type: "conv" type: "conv"
name: "___mixed_6__.w1" name: "___mixed_6__.w1"
...@@ -177,6 +178,7 @@ layers { ...@@ -177,6 +178,7 @@ layers {
output_y: 30 output_y: 30
img_size_y: 32 img_size_y: 32
} }
num_filters: 64
} }
} }
inputs { inputs {
...@@ -218,6 +220,7 @@ layers { ...@@ -218,6 +220,7 @@ layers {
} }
inputs { inputs {
input_layer_name: "img" input_layer_name: "img"
input_parameter_name: "___mixed_7__.w1"
proj_conf { proj_conf {
type: "convt" type: "convt"
name: "___mixed_7__.w1" name: "___mixed_7__.w1"
...@@ -239,6 +242,7 @@ layers { ...@@ -239,6 +242,7 @@ layers {
output_y: 32 output_y: 32
img_size_y: 63 img_size_y: 63
} }
num_filters: 64
} }
} }
inputs { inputs {
...@@ -377,6 +381,22 @@ parameters { ...@@ -377,6 +381,22 @@ parameters {
initial_strategy: 0 initial_strategy: 0
initial_smart: true initial_smart: true
} }
parameters {
name: "___mixed_6__.w1"
size: 576
initial_mean: 0.0
initial_std: 0.471404520791
initial_strategy: 0
initial_smart: false
}
parameters {
name: "___mixed_7__.w1"
size: 576
initial_mean: 0.0
initial_std: 0.471404520791
initial_strategy: 0
initial_smart: false
}
parameters { parameters {
name: "___mixed_8__.w0" name: "___mixed_8__.w0"
size: 30000 size: 30000
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册