From e651f3f62fd843759cdaa237ba724593fe900e1c Mon Sep 17 00:00:00 2001 From: weishengyu Date: Fri, 8 Apr 2022 10:58:23 +0800 Subject: [PATCH] remove bnneck --- ppcls/arch/gears/bnneck.py | 2 +- ppcls/configs/Pedestrian/strong_baseline_m1.yaml | 8 ++++---- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/ppcls/arch/gears/bnneck.py b/ppcls/arch/gears/bnneck.py index 15b6f46b..2e4779e2 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 718d5761..7edb9349 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 -- GitLab