Update BNNeck

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()
 

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