diff --git a/ppcls/arch/gears/bnneck.py b/ppcls/arch/gears/bnneck.py index 15b6f46b05a7c0390257b1882d006612cef4390d..2e4779e2ab369b718002376327bb360530e0971e 100644 --- a/ppcls/arch/gears/bnneck.py +++ b/ppcls/arch/gears/bnneck.py @@ -6,7 +6,7 @@ class BNNeck(paddle.nn.Layer): super(BNNeck, self).__init__() self.num_filters = num_filters - self.bn = paddle.nn.BatchNorm( + self.bn = paddle.nn.BatchNorm1D( self.num_filters) if not trainable: self.bn.weight.trainable = False diff --git a/ppcls/configs/Pedestrian/strong_baseline_m1.yaml b/ppcls/configs/Pedestrian/strong_baseline_m1.yaml index 718d5761bfe92783a844863258fa2312e9a5683c..7edb9349b6760d67dece7446f46aa0aff9694526 100644 --- a/ppcls/configs/Pedestrian/strong_baseline_m1.yaml +++ b/ppcls/configs/Pedestrian/strong_baseline_m1.yaml @@ -26,10 +26,10 @@ Arch: stem_act: null BackboneStopLayer: name: "flatten" - Neck: - name: BNNeck - num_filters: 2048 - trainable: false + #Neck: + # name: BNNeck + # num_filters: 2048 + # trainable: false Head: name: "FC" embedding_size: 2048