dbg

ppcls/arch/backbone/legendary_models/resnet.py

@@ -280,6 +280,7 @@ class ResNet(TheseusLayer):
                  lr_mult_list=[1.0, 1.0, 1.0, 1.0, 1.0],
                  data_format="NCHW",
                  input_image_channel=3,
+                 stem_act="relu",
                  return_patterns=None,
                  return_stages=None):
         super().__init__()
@@ -315,7 +316,7 @@ class ResNet(TheseusLayer):
                 num_filters=out_c,
                 filter_size=k,
                 stride=s,
-                act="relu",
+                act=stem_act,
                 lr_mult=self.lr_mult_list[0],
                 data_format=data_format)
             for in_c, out_c, k, s in self.stem_cfg[version]

ppcls/configs/Pedestrian/reid_sb_ResNet50.yaml

@@ -23,6 +23,7 @@ Arch:
   Backbone:
     name: "ResNet50"
     pretrained: True
+    stem_act: null
   BackboneStopLayer:
     name: "flatten"
   Head:
@@ -68,13 +69,13 @@ DataLoader:
               to_rgb: True
               channel_first: False
           - ResizeImage:
-              size: [256, 128]
+              size: [128, 256]
           - RandFlipImage:
               flip_code: 1
           - Pad:
               padding: 10
           - RandCropImage:
-              size: [256, 128]
+              size: [128, 256]
           - NormalizeImage:
               scale: 0.00392157
               mean: [0.485, 0.456, 0.406]
@@ -88,7 +89,7 @@ DataLoader:
               mean: [0.4914, 0.4822, 0.4465]
     sampler:
         name: PKSampler
-        batch_size: 128
+        batch_size: 64
         sample_per_id: 4
         drop_last: True
         shuffle: True
@@ -106,7 +107,7 @@ DataLoader:
               to_rgb: True
               channel_first: False
           - ResizeImage:
-              size: [256, 128]
+              size: [128, 256]
           - NormalizeImage:
               scale: 0.00392157
               mean: [0.485, 0.456, 0.406]
@@ -114,7 +115,7 @@ DataLoader:
               order: ''
       sampler:
         name: DistributedBatchSampler
-        batch_size: 64
+        batch_size: 128
         drop_last: False
         shuffle: False
       loader:
@@ -131,7 +132,7 @@ DataLoader:
               to_rgb: True
               channel_first: False
           - ResizeImage:
-              size: [256, 128]
+              size: [128, 256]
           - NormalizeImage:
               scale: 0.00392157
               mean: [0.485, 0.456, 0.406]
@@ -139,7 +140,7 @@ DataLoader:
               order: ''
       sampler:
         name: DistributedBatchSampler
-        batch_size: 64
+        batch_size: 128
         drop_last: False
         shuffle: False
       loader: