From 1be9c16a9537c31524fcf576a83ea2a9c2788643 Mon Sep 17 00:00:00 2001
From: tianyi1997 <93087391+tianyi1997@users.noreply.github.com>
Date: Wed, 8 Feb 2023 15:22:37 +0800
Subject: [PATCH] Update BNNeck

---
 ppcls/arch/gears/bnneck.py                             | 7 ++++++-
 ppcls/configs/reid/MetaBIN_ResNet50_single_source.yaml | 1 +
 2 files changed, 7 insertions(+), 1 deletion(-)

diff --git a/ppcls/arch/gears/bnneck.py b/ppcls/arch/gears/bnneck.py
index 916a612a..e7abb892 100644
--- a/ppcls/arch/gears/bnneck.py
+++ b/ppcls/arch/gears/bnneck.py
@@ -36,12 +36,17 @@ class BNNeck(nn.Layer):
         if 'bias_attr' in kwargs:
             bias_attr = get_param_attr_dict(kwargs['bias_attr'])
 
+        use_global_stats = None
+        if 'use_global_stats' in kwargs:
+            use_global_stats = get_param_attr_dict(kwargs['use_global_stats'])
+
         self.feat_bn = nn.BatchNorm1D(
             num_features,
             momentum=0.9,
             epsilon=1e-05,
             weight_attr=weight_attr,
-            bias_attr=bias_attr)
+            bias_attr=bias_attr,
+            use_global_stats=use_global_stats)
 
         self.flatten = nn.Flatten()
 
diff --git a/ppcls/configs/reid/MetaBIN_ResNet50_single_source.yaml b/ppcls/configs/reid/MetaBIN_ResNet50_single_source.yaml
index 1e1874dd..8b301146 100644
--- a/ppcls/configs/reid/MetaBIN_ResNet50_single_source.yaml
+++ b/ppcls/configs/reid/MetaBIN_ResNet50_single_source.yaml
@@ -36,6 +36,7 @@ Arch:
   Neck:
     name: BNNeck
     num_features: &feat_dim 2048
+    use_global_stats: True
   Head:
     name: "FC"
     embedding_size: *feat_dim
-- 
GitLab