Skip to content

Commit e9d839d

Browse files
authored
Merge pull request #1693 from wangyang59/cudnnDeconv
fixed a bug in conv_projection issue #1681
2 parents 32667db + dc45cb4 commit e9d839d

File tree

3 files changed

+25
-2
lines changed

3 files changed

+25
-2
lines changed

.gitignore

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,7 @@ build/
77
.project
88
.cproject
99
.pydevproject
10+
.settings/
1011
Makefile
1112
.test_env/
1213
third_party/

python/paddle/trainer/config_parser.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -724,7 +724,8 @@ def __init__(self,
724724
num_filters=None,
725725
conv_conf=None,
726726
**xargs):
727-
super(ConvProjection, self).__init__(input_layer_name, **xargs)
727+
super(ConvProjection, self).__init__(input_layer_name, num_filters,
728+
conv_conf, **xargs)
728729

729730
parse_conv(conv_conf, self.input_layer_name, self.proj_conf.conv_conf,
730731
num_filters)
@@ -742,7 +743,8 @@ def __init__(self,
742743
num_filters=None,
743744
conv_conf=None,
744745
**xargs):
745-
super(ConvTransProjection, self).__init__(input_layer_name, **xargs)
746+
super(ConvTransProjection, self).__init__(input_layer_name, num_filters,
747+
conv_conf, **xargs)
746748

747749
parse_conv(
748750
conv_conf,

python/paddle/trainer_config_helpers/tests/configs/protostr/projections.protostr

Lines changed: 20 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -156,6 +156,7 @@ layers {
156156
}
157157
inputs {
158158
input_layer_name: "img"
159+
input_parameter_name: "___mixed_6__.w1"
159160
proj_conf {
160161
type: "conv"
161162
name: "___mixed_6__.w1"
@@ -177,6 +178,7 @@ layers {
177178
output_y: 30
178179
img_size_y: 32
179180
}
181+
num_filters: 64
180182
}
181183
}
182184
inputs {
@@ -218,6 +220,7 @@ layers {
218220
}
219221
inputs {
220222
input_layer_name: "img"
223+
input_parameter_name: "___mixed_7__.w1"
221224
proj_conf {
222225
type: "convt"
223226
name: "___mixed_7__.w1"
@@ -239,6 +242,7 @@ layers {
239242
output_y: 32
240243
img_size_y: 63
241244
}
245+
num_filters: 64
242246
}
243247
}
244248
inputs {
@@ -377,6 +381,22 @@ parameters {
377381
initial_strategy: 0
378382
initial_smart: true
379383
}
384+
parameters {
385+
name: "___mixed_6__.w1"
386+
size: 576
387+
initial_mean: 0.0
388+
initial_std: 0.471404520791
389+
initial_strategy: 0
390+
initial_smart: false
391+
}
392+
parameters {
393+
name: "___mixed_7__.w1"
394+
size: 576
395+
initial_mean: 0.0
396+
initial_std: 0.471404520791
397+
initial_strategy: 0
398+
initial_smart: false
399+
}
380400
parameters {
381401
name: "___mixed_8__.w0"
382402
size: 30000

0 commit comments

Comments
 (0)