Skip to content

Commit 62c5389

Browse files
committed
fix test_layerHelpers unittest error
1 parent ca0bb40 commit 62c5389

File tree

1 file changed

+3
-4
lines changed
  • python/paddle/trainer_config_helpers

1 file changed

+3
-4
lines changed

python/paddle/trainer_config_helpers/layers.py

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1827,7 +1827,6 @@ def img_pool_layer(input, pool_size, name=None,
18271827
@layer_support()
18281828
def spp_layer(input, name=None, num_channels=None, pool_type=None,
18291829
pyramid_height=None, img_width=None, layer_attr=None):
1830-
pass
18311830
"""
18321831
Spatial Pyramid Pooling in Deep Convolutional Networks for Visual Recognition.
18331832
The details please refer to
@@ -1864,7 +1863,7 @@ def spp_layer(input, name=None, num_channels=None, pool_type=None,
18641863
if (isinstance(pool_type, AvgPooling) or isinstance(pool_type, MaxPooling)):
18651864
type_name += '-projection'
18661865

1867-
Layer(
1866+
l = Layer(
18681867
name=name,
18691868
type=LayerType.SPP_LAYER,
18701869
inputs=Input(input.name,
@@ -1875,8 +1874,8 @@ def spp_layer(input, name=None, num_channels=None, pool_type=None,
18751874
),
18761875
**ExtraLayerAttribute.to_kwargs(layer_attr)
18771876
)
1878-
return LayerOutput(name, LayerType.SPP_LAYER, parents=[input],
1879-
num_filters=num_channels)
1877+
return LayerOutput(name, layer_type=LayerType.SPP_LAYER, parents=[input],
1878+
num_filters=num_channels, size=l.config.size)
18801879

18811880

18821881
def __img_norm_layer__(name, input, size, norm_type, scale, power,

0 commit comments

Comments
 (0)