Skip to content

Commit 2ab928d

Browse files
committed
modify the prelu test and regenerate the proto
1 parent f9469d3 commit 2ab928d

File tree

2 files changed

+84
-15
lines changed

2 files changed

+84
-15
lines changed

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

Lines changed: 78 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,8 @@ layers {
44
type: "data"
55
size: 300
66
active_type: ""
7+
height: 10
8+
width: 10
79
}
810
layers {
911
name: "__prelu_layer_0__"
@@ -15,6 +17,9 @@ layers {
1517
input_parameter_name: "___prelu_layer_0__.w0"
1618
}
1719
partial_sum: 1
20+
height: 10
21+
width: 10
22+
depth: 1
1823
}
1924
layers {
2025
name: "__prelu_layer_1__"
@@ -26,6 +31,9 @@ layers {
2631
input_parameter_name: "___prelu_layer_1__.w0"
2732
}
2833
partial_sum: 1
34+
height: 10
35+
width: 10
36+
depth: 1
2937
}
3038
layers {
3139
name: "__prelu_layer_2__"
@@ -37,41 +45,100 @@ layers {
3745
input_parameter_name: "___prelu_layer_2__.w0"
3846
}
3947
partial_sum: 5
48+
height: 10
49+
width: 10
50+
depth: 1
51+
}
52+
layers {
53+
name: "__prelu_layer_3__"
54+
type: "prelu"
55+
size: 300
56+
active_type: ""
57+
inputs {
58+
input_layer_name: "input"
59+
input_parameter_name: "___prelu_layer_3__.w0"
60+
}
61+
partial_sum: 300
62+
height: 10
63+
width: 10
64+
depth: 1
65+
}
66+
layers {
67+
name: "__prelu_layer_4__"
68+
type: "prelu"
69+
size: 300
70+
active_type: ""
71+
inputs {
72+
input_layer_name: "input"
73+
input_parameter_name: "___prelu_layer_4__.w0"
74+
}
75+
partial_sum: 100
76+
height: 10
77+
width: 10
78+
depth: 1
4079
}
4180
parameters {
4281
name: "___prelu_layer_0__.w0"
4382
size: 300
44-
initial_mean: 0.0
45-
initial_std: 0.057735026919
83+
initial_mean: 0.25
84+
initial_std: 0.0
85+
dims: 1
86+
dims: 300
4687
initial_strategy: 0
47-
initial_smart: true
88+
initial_smart: false
4889
}
4990
parameters {
5091
name: "___prelu_layer_1__.w0"
5192
size: 300
52-
initial_mean: 0.0
53-
initial_std: 0.057735026919
93+
initial_mean: 0.25
94+
initial_std: 0.0
95+
dims: 1
96+
dims: 300
5497
initial_strategy: 0
55-
initial_smart: true
98+
initial_smart: false
5699
}
57100
parameters {
58101
name: "___prelu_layer_2__.w0"
59102
size: 60
60-
initial_mean: 0.0
61-
initial_std: 0.129099444874
103+
initial_mean: 0.25
104+
initial_std: 0.0
105+
dims: 1
106+
dims: 60
107+
initial_strategy: 0
108+
initial_smart: false
109+
}
110+
parameters {
111+
name: "___prelu_layer_3__.w0"
112+
size: 1
113+
initial_mean: 0.25
114+
initial_std: 0.0
115+
dims: 1
116+
dims: 1
117+
initial_strategy: 0
118+
initial_smart: false
119+
}
120+
parameters {
121+
name: "___prelu_layer_4__.w0"
122+
size: 3
123+
initial_mean: 0.25
124+
initial_std: 0.0
125+
dims: 1
126+
dims: 3
62127
initial_strategy: 0
63-
initial_smart: true
128+
initial_smart: false
64129
}
65130
input_layer_names: "input"
66-
output_layer_names: "__prelu_layer_2__"
131+
output_layer_names: "__prelu_layer_4__"
67132
sub_models {
68133
name: "root"
69134
layer_names: "input"
70135
layer_names: "__prelu_layer_0__"
71136
layer_names: "__prelu_layer_1__"
72137
layer_names: "__prelu_layer_2__"
138+
layer_names: "__prelu_layer_3__"
139+
layer_names: "__prelu_layer_4__"
73140
input_layer_names: "input"
74-
output_layer_names: "__prelu_layer_2__"
141+
output_layer_names: "__prelu_layer_4__"
75142
is_recurrent_layer_group: false
76143
}
77144

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,8 +1,10 @@
11
from paddle.trainer_config_helpers import *
22

3-
data = data_layer(name='input', size=300)
4-
prelu = prelu_layer(input=data)
5-
prelu = prelu_layer(input=data, partial_sum=1)
6-
prelu = prelu_layer(input=data, partial_sum=5)
3+
data = data_layer(name='input', size=300, height=10, width=10)
4+
prelu = prelu_layer(input=data, num_channels=3)
5+
prelu = prelu_layer(input=data, partial_sum=1, num_channels=3)
6+
prelu = prelu_layer(input=data, partial_sum=5, num_channels=3)
7+
prelu = prelu_layer(input=data, channel_shared=True, num_channels=3)
8+
prelu = prelu_layer(input=data, channel_shared=False, num_channels=3)
79

810
outputs(prelu)

0 commit comments

Comments
 (0)