fix

ppcls/configs/PULC/safety_helmet/MobileNetV3_small_x0_35.yaml

@@ -7,7 +7,7 @@ Global:
   save_interval: 1
   eval_during_train: True
   eval_interval: 1
-  epochs: 40
+  epochs: 60
   print_batch_step: 10
   use_visualdl: False
   # used for static mode and model export

ppcls/configs/PULC/safety_helmet/PPLCNet_x1_0.yaml

@@ -13,12 +13,14 @@ Global:
   # used for static mode and model export
   image_shape: [3, 224, 224]
   save_inference_dir: ./inference
+
 # model architecture
 Arch:
   name: PPLCNet_x1_0
   pretrained: True
   use_ssld: True
   class_num: 2
+  use_sync_bn : True
  
 # loss function config for traing/eval process
 Loss:
@@ -36,7 +38,7 @@ Optimizer:
   momentum: 0.9
   lr:
     name: Cosine
-    learning_rate: 0.01
+    learning_rate: 0.025
     warmup_epoch: 5
   regularizer:
     name: 'L2'
@@ -55,15 +57,27 @@ DataLoader:
             to_rgb: True
             channel_first: False
         - RandCropImage:
-            size: 224
+            size: 176
         - RandFlipImage:
             flip_code: 1
+        - TimmAutoAugment:
+            prob : 0.5
+            config_str: rand-m9-mstd0.5-inc1
+            interpolation: bicubic
+            img_size : 176
         - NormalizeImage:
             scale: 1.0/255.0
             mean: [0.485, 0.456, 0.406]
             std: [0.229, 0.224, 0.225]
             order: ''
-
+        - RandomErasing:
+            EPSILON : 0.1
+            r1 : 0.3
+            sh : 1.0/3.0
+            sl : 0.02
+            attempt : 10
+            use_log_aspect : True
+            mode : pixel
     sampler:
       name: DistributedBatchSampler
       batch_size: 64

ppcls/configs/PULC/safety_helmet/PPLCNet_x1_0_search.yaml

@@ -37,7 +37,7 @@ Optimizer:
   momentum: 0.9
   lr:
     name: Cosine
-    learning_rate: 0.015
+    learning_rate: 0.10
     warmup_epoch: 5
   regularizer:
     name: 'L2'
@@ -60,7 +60,7 @@ DataLoader:
         - RandFlipImage:
             flip_code: 1
         - TimmAutoAugment:
-            prob: 0.5
+            prob: 0
             config_str: rand-m9-mstd0.5-inc1
             interpolation: bicubic
             img_size: 192
@@ -70,7 +70,7 @@ DataLoader:
             std: [0.229, 0.224, 0.225]
             order: ''
         - RandomErasing:
-            EPSILON: 0.5
+            EPSILON: 0
             sl: 0.02
             sh: 1.0/3.0
             r1: 0.3

ppcls/configs/PULC/safety_helmet/Res2Net200_vd_26w_4s.yaml

@@ -7,7 +7,7 @@ Global:
   save_interval: 1
   eval_during_train: True
   eval_interval: 1
-  epochs: 40
+  epochs: 60
   print_batch_step: 10
   use_visualdl: False
   # used for static mode and model export

ppcls/configs/PULC/safety_helmet/SwinTransformer_tiny_patch4_window7_224.yaml

@@ -7,7 +7,7 @@ Global:
   save_interval: 1
   eval_during_train: True
   eval_interval: 1
-  epochs: 40
+  epochs: 60
   print_batch_step: 10
   use_visualdl: False
   # used for static mode and model export
@@ -81,6 +81,14 @@ DataLoader:
             attempt: 10
             use_log_aspect: True
             mode: pixel
+      batch_transform_ops:
+        - OpSampler:
+            MixupOperator:
+              alpha: 0.8
+              prob: 0.5
+            CutmixOperator:
+              alpha: 1.0
+              prob: 0.5
 
     sampler:
       name: DistributedBatchSampler

ppcls/configs/PULC/safety_helmet/search.yaml

@@ -8,7 +8,7 @@ search_dict:
   - search_key: lrs
     replace_config:
       - Optimizer.lr.learning_rate
-    search_values: [0.01, 0.015, 0.025]
+    search_values: [0.03, 0.04, 0.05, 0.06, 0.07, 0.08, 0.09, 0.10, 0.11, 0.11, 0.12]
   - search_key: resolutions
     replace_config:
       - DataLoader.Train.dataset.transform_ops.1.RandCropImage.size