Update configs

ppcls/configs/ImageNet/HRNet/HRNet_W18_C.yaml

@@ -30,10 +30,10 @@ Loss:
 
 
 Optimizer:
-  name: Momentum
+  name: "Momentum"
   momentum: 0.9
   lr:
-    name: Piecewise
+    name: "Piecewise"
     learning_rate: 0.1
     decay_epochs: [30, 60, 90]
     values: [0.1, 0.01, 0.001, 0.0001]
@@ -46,80 +46,80 @@ Optimizer:
 DataLoader:
   Train:
     dataset:
-        name: ImageNetDataset
-        image_root: "./dataset/ILSVRC2012/"
-        cls_label_path: "./dataset/ILSVRC2012/train_list.txt"
-        transform_ops:
-          - RandCropImage:
-              size: 224
-          - RandFlipImage:
-              flip_code: 1
-          - NormalizeImage:
-              scale: 0.00392157
-              mean: [0.485, 0.456, 0.406]
-              std: [0.229, 0.224, 0.225]
-              order: ''
+      name: "ImageNetDataset"
+      image_root: "./dataset/ILSVRC2012/"
+      cls_label_path: "./dataset/ILSVRC2012/train_list.txt"
+      transform_ops:
+        - RandCropImage:
+            size: 224
+        - RandFlipImage:
+            flip_code: 1
+        - NormalizeImage:
+            scale: 1.0/255.0
+            mean: [0.485, 0.456, 0.406]
+            std: [0.229, 0.224, 0.225]
+            order: ''
 
     sampler:
-        name: DistributedBatchSampler
-        batch_size: 64
-        drop_last: False
-        shuffle: True
+      name: "DistributedBatchSampler"
+      batch_size: 64
+      drop_last: False
+      shuffle: True
     loader:
-        num_workers: 6
-        use_shared_memory: True
+      num_workers: 4
+      use_shared_memory: True
 
   Eval:
     # TOTO: modify to the latest trainer
     dataset: 
-        name: ImageNetDataset
-        image_root: "./dataset/ILSVRC2012/"
-        cls_label_path: "./dataset/ILSVRC2012/val_list.txt"
-        transform_ops:
-          - ResizeImage:
-              resize_short: 256
-          - CropImage:
-              size: 224
-          - NormalizeImage:
-              scale: 0.00392157
-              mean: [0.485, 0.456, 0.406]
-              std: [0.229, 0.224, 0.225]
-              order: ''
+      name: "ImageNetDataset"
+      image_root: "./dataset/ILSVRC2012/"
+      cls_label_path: "./dataset/ILSVRC2012/val_list.txt"
+      transform_ops:
+        - ResizeImage:
+            resize_short: 256
+        - CropImage:
+            size: 224
+        - NormalizeImage:
+            scale: 1.0/255.0
+            mean: [0.485, 0.456, 0.406]
+            std: [0.229, 0.224, 0.225]
+            order: ''
     sampler:
-        name: DistributedBatchSampler
-        batch_size: 64
-        drop_last: False
-        shuffle: False
+      name: "DistributedBatchSampler"
+      batch_size: 64
+      drop_last: False
+      shuffle: False
     loader:
-        num_workers: 6
-        use_shared_memory: True
+      num_workers: 4
+      use_shared_memory: True
 
 Infer:
   infer_imgs: "docs/images/whl/demo.jpg"
   batch_size: 10
   transforms:
-      - DecodeImage:
-          to_rgb: True
-          channel_first: False
-      - ResizeImage:
-          resize_short: 256
-      - CropImage:
-          size: 224
-      - NormalizeImage:
-          scale: 1.0/255.0
-          mean: [0.485, 0.456, 0.406]
-          std: [0.229, 0.224, 0.225]
-          order: ''
-      - ToCHWImage:
+    - DecodeImage:
+        to_rgb: True
+        channel_first: False
+    - ResizeImage:
+        resize_short: 256
+    - CropImage:
+        size: 224
+    - NormalizeImage:
+        scale: 1.0/255.0
+        mean: [0.485, 0.456, 0.406]
+        std: [0.229, 0.224, 0.225]
+        order: ''
+    - ToCHWImage:
   PostProcess:
-    name: Topk
+    name: "Topk"
     topk: 5
     class_id_map_file: "ppcls/utils/imagenet1k_label_list.txt"
 
 Metric:
-    Train:
+  Train:
     - TopkAcc:
         topk: [1, 5]
-    Eval:
+  Eval:
     - TopkAcc:
         topk: [1, 5]

ppcls/configs/ImageNet/HRNet/HRNet_W30_C.yaml

@@ -30,10 +30,10 @@ Loss:
 
 
 Optimizer:
-  name: Momentum
+  name: "Momentum"
   momentum: 0.9
   lr:
-    name: Piecewise
+    name: "Piecewise"
     learning_rate: 0.1
     decay_epochs: [30, 60, 90]
     values: [0.1, 0.01, 0.001, 0.0001]
@@ -46,80 +46,80 @@ Optimizer:
 DataLoader:
   Train:
     dataset:
-        name: ImageNetDataset
-        image_root: "./dataset/ILSVRC2012/"
-        cls_label_path: "./dataset/ILSVRC2012/train_list.txt"
-        transform_ops:
-          - RandCropImage:
-              size: 224
-          - RandFlipImage:
-              flip_code: 1
-          - NormalizeImage:
-              scale: 0.00392157
-              mean: [0.485, 0.456, 0.406]
-              std: [0.229, 0.224, 0.225]
-              order: ''
+      name: "ImageNetDataset"
+      image_root: "./dataset/ILSVRC2012/"
+      cls_label_path: "./dataset/ILSVRC2012/train_list.txt"
+      transform_ops:
+        - RandCropImage:
+            size: 224
+        - RandFlipImage:
+            flip_code: 1
+        - NormalizeImage:
+            scale: 1.0/255.0
+            mean: [0.485, 0.456, 0.406]
+            std: [0.229, 0.224, 0.225]
+            order: ''
 
     sampler:
-        name: DistributedBatchSampler
-        batch_size: 64
-        drop_last: False
-        shuffle: True
+      name: "DistributedBatchSampler"
+      batch_size: 64
+      drop_last: False
+      shuffle: True
     loader:
-        num_workers: 6
-        use_shared_memory: True
+      num_workers: 4
+      use_shared_memory: True
 
   Eval:
     # TOTO: modify to the latest trainer
     dataset: 
-        name: ImageNetDataset
-        image_root: "./dataset/ILSVRC2012/"
-        cls_label_path: "./dataset/ILSVRC2012/val_list.txt"
-        transform_ops:
-          - ResizeImage:
-              resize_short: 256
-          - CropImage:
-              size: 224
-          - NormalizeImage:
-              scale: 0.00392157
-              mean: [0.485, 0.456, 0.406]
-              std: [0.229, 0.224, 0.225]
-              order: ''
+      name: "ImageNetDataset"
+      image_root: "./dataset/ILSVRC2012/"
+      cls_label_path: "./dataset/ILSVRC2012/val_list.txt"
+      transform_ops:
+        - ResizeImage:
+            resize_short: 256
+        - CropImage:
+            size: 224
+        - NormalizeImage:
+            scale: 1.0/255.0
+            mean: [0.485, 0.456, 0.406]
+            std: [0.229, 0.224, 0.225]
+            order: ''
     sampler:
-        name: DistributedBatchSampler
-        batch_size: 64
-        drop_last: False
-        shuffle: False
+      name: "DistributedBatchSampler"
+      batch_size: 64
+      drop_last: False
+      shuffle: False
     loader:
-        num_workers: 6
-        use_shared_memory: True
+      num_workers: 4
+      use_shared_memory: True
 
 Infer:
   infer_imgs: "docs/images/whl/demo.jpg"
   batch_size: 10
   transforms:
-      - DecodeImage:
-          to_rgb: True
-          channel_first: False
-      - ResizeImage:
-          resize_short: 256
-      - CropImage:
-          size: 224
-      - NormalizeImage:
-          scale: 1.0/255.0
-          mean: [0.485, 0.456, 0.406]
-          std: [0.229, 0.224, 0.225]
-          order: ''
-      - ToCHWImage:
+    - DecodeImage:
+        to_rgb: True
+        channel_first: False
+    - ResizeImage:
+        resize_short: 256
+    - CropImage:
+        size: 224
+    - NormalizeImage:
+        scale: 1.0/255.0
+        mean: [0.485, 0.456, 0.406]
+        std: [0.229, 0.224, 0.225]
+        order: ''
+    - ToCHWImage:
   PostProcess:
-    name: Topk
+    name: "Topk"
     topk: 5
     class_id_map_file: "ppcls/utils/imagenet1k_label_list.txt"
 
 Metric:
-    Train:
+  Train:
     - TopkAcc:
         topk: [1, 5]
-    Eval:
+  Eval:
     - TopkAcc:
         topk: [1, 5]

ppcls/configs/ImageNet/HRNet/HRNet_W32_C.yaml

@@ -30,10 +30,10 @@ Loss:
 
 
 Optimizer:
-  name: Momentum
+  name: "Momentum"
   momentum: 0.9
   lr:
-    name: Piecewise
+    name: "Piecewise"
     learning_rate: 0.1
     decay_epochs: [30, 60, 90]
     values: [0.1, 0.01, 0.001, 0.0001]
@@ -46,80 +46,80 @@ Optimizer:
 DataLoader:
   Train:
     dataset:
-        name: ImageNetDataset
-        image_root: "./dataset/ILSVRC2012/"
-        cls_label_path: "./dataset/ILSVRC2012/train_list.txt"
-        transform_ops:
-          - RandCropImage:
-              size: 224
-          - RandFlipImage:
-              flip_code: 1
-          - NormalizeImage:
-              scale: 0.00392157
-              mean: [0.485, 0.456, 0.406]
-              std: [0.229, 0.224, 0.225]
-              order: ''
+      name: "ImageNetDataset"
+      image_root: "./dataset/ILSVRC2012/"
+      cls_label_path: "./dataset/ILSVRC2012/train_list.txt"
+      transform_ops:
+        - RandCropImage:
+            size: 224
+        - RandFlipImage:
+            flip_code: 1
+        - NormalizeImage:
+            scale: 1.0/255.0
+            mean: [0.485, 0.456, 0.406]
+            std: [0.229, 0.224, 0.225]
+            order: ''
 
     sampler:
-        name: DistributedBatchSampler
-        batch_size: 64
-        drop_last: False
-        shuffle: True
+      name: "DistributedBatchSampler"
+      batch_size: 64
+      drop_last: False
+      shuffle: True
     loader:
-        num_workers: 6
-        use_shared_memory: True
+      num_workers: 4
+      use_shared_memory: True
 
   Eval:
     # TOTO: modify to the latest trainer
     dataset: 
-        name: ImageNetDataset
-        image_root: "./dataset/ILSVRC2012/"
-        cls_label_path: "./dataset/ILSVRC2012/val_list.txt"
-        transform_ops:
-          - ResizeImage:
-              resize_short: 256
-          - CropImage:
-              size: 224
-          - NormalizeImage:
-              scale: 0.00392157
-              mean: [0.485, 0.456, 0.406]
-              std: [0.229, 0.224, 0.225]
-              order: ''
+      name: "ImageNetDataset"
+      image_root: "./dataset/ILSVRC2012/"
+      cls_label_path: "./dataset/ILSVRC2012/val_list.txt"
+      transform_ops:
+        - ResizeImage:
+            resize_short: 256
+        - CropImage:
+            size: 224
+        - NormalizeImage:
+            scale: 1.0/255.0
+            mean: [0.485, 0.456, 0.406]
+            std: [0.229, 0.224, 0.225]
+            order: ''
     sampler:
-        name: DistributedBatchSampler
-        batch_size: 64
-        drop_last: False
-        shuffle: False
+      name: "DistributedBatchSampler"
+      batch_size: 64
+      drop_last: False
+      shuffle: False
     loader:
-        num_workers: 6
-        use_shared_memory: True
+      num_workers: 4
+      use_shared_memory: True
 
 Infer:
   infer_imgs: "docs/images/whl/demo.jpg"
   batch_size: 10
   transforms:
-      - DecodeImage:
-          to_rgb: True
-          channel_first: False
-      - ResizeImage:
-          resize_short: 256
-      - CropImage:
-          size: 224
-      - NormalizeImage:
-          scale: 1.0/255.0
-          mean: [0.485, 0.456, 0.406]
-          std: [0.229, 0.224, 0.225]
-          order: ''
-      - ToCHWImage:
+    - DecodeImage:
+        to_rgb: True
+        channel_first: False
+    - ResizeImage:
+        resize_short: 256
+    - CropImage:
+        size: 224
+    - NormalizeImage:
+        scale: 1.0/255.0
+        mean: [0.485, 0.456, 0.406]
+        std: [0.229, 0.224, 0.225]
+        order: ''
+    - ToCHWImage:
   PostProcess:
-    name: Topk
+    name: "Topk"
     topk: 5
     class_id_map_file: "ppcls/utils/imagenet1k_label_list.txt"
 
 Metric:
-    Train:
+  Train:
     - TopkAcc:
         topk: [1, 5]
-    Eval:
+  Eval:
     - TopkAcc:
         topk: [1, 5]

ppcls/configs/ImageNet/HRNet/HRNet_W40_C.yaml

@@ -30,10 +30,10 @@ Loss:
 
 
 Optimizer:
-  name: Momentum
+  name: "Momentum"
   momentum: 0.9
   lr:
-    name: Piecewise
+    name: "Piecewise"
     learning_rate: 0.1
     decay_epochs: [30, 60, 90]
     values: [0.1, 0.01, 0.001, 0.0001]
@@ -46,80 +46,80 @@ Optimizer:
 DataLoader:
   Train:
     dataset:
-        name: ImageNetDataset
-        image_root: "./dataset/ILSVRC2012/"
-        cls_label_path: "./dataset/ILSVRC2012/train_list.txt"
-        transform_ops:
-          - RandCropImage:
-              size: 224
-          - RandFlipImage:
-              flip_code: 1
-          - NormalizeImage:
-              scale: 0.00392157
-              mean: [0.485, 0.456, 0.406]
-              std: [0.229, 0.224, 0.225]
-              order: ''
+      name: "ImageNetDataset"
+      image_root: "./dataset/ILSVRC2012/"
+      cls_label_path: "./dataset/ILSVRC2012/train_list.txt"
+      transform_ops:
+        - RandCropImage:
+            size: 224
+        - RandFlipImage:
+            flip_code: 1
+        - NormalizeImage:
+            scale: 1.0/255.0
+            mean: [0.485, 0.456, 0.406]
+            std: [0.229, 0.224, 0.225]
+            order: ''
 
     sampler:
-        name: DistributedBatchSampler
-        batch_size: 64
-        drop_last: False
-        shuffle: True
+      name: "DistributedBatchSampler"
+      batch_size: 64
+      drop_last: False
+      shuffle: True
     loader:
-        num_workers: 6
-        use_shared_memory: True
+      num_workers: 4
+      use_shared_memory: True
 
   Eval:
     # TOTO: modify to the latest trainer
     dataset: 
-        name: ImageNetDataset
-        image_root: "./dataset/ILSVRC2012/"
-        cls_label_path: "./dataset/ILSVRC2012/val_list.txt"
-        transform_ops:
-          - ResizeImage:
-              resize_short: 256
-          - CropImage:
-              size: 224
-          - NormalizeImage:
-              scale: 0.00392157
-              mean: [0.485, 0.456, 0.406]
-              std: [0.229, 0.224, 0.225]
-              order: ''
+      name: "ImageNetDataset"
+      image_root: "./dataset/ILSVRC2012/"
+      cls_label_path: "./dataset/ILSVRC2012/val_list.txt"
+      transform_ops:
+        - ResizeImage:
+            resize_short: 256
+        - CropImage:
+            size: 224
+        - NormalizeImage:
+            scale: 1.0/255.0
+            mean: [0.485, 0.456, 0.406]
+            std: [0.229, 0.224, 0.225]
+            order: ''
     sampler:
-        name: DistributedBatchSampler
-        batch_size: 64
-        drop_last: False
-        shuffle: False
+      name: "DistributedBatchSampler"
+      batch_size: 64
+      drop_last: False
+      shuffle: False
     loader:
-        num_workers: 6
-        use_shared_memory: True
+      num_workers: 4
+      use_shared_memory: True
 
 Infer:
   infer_imgs: "docs/images/whl/demo.jpg"
   batch_size: 10
   transforms:
-      - DecodeImage:
-          to_rgb: True
-          channel_first: False
-      - ResizeImage:
-          resize_short: 256
-      - CropImage:
-          size: 224
-      - NormalizeImage:
-          scale: 1.0/255.0
-          mean: [0.485, 0.456, 0.406]
-          std: [0.229, 0.224, 0.225]
-          order: ''
-      - ToCHWImage:
+    - DecodeImage:
+        to_rgb: True
+        channel_first: False
+    - ResizeImage:
+        resize_short: 256
+    - CropImage:
+        size: 224
+    - NormalizeImage:
+        scale: 1.0/255.0
+        mean: [0.485, 0.456, 0.406]
+        std: [0.229, 0.224, 0.225]
+        order: ''
+    - ToCHWImage:
   PostProcess:
-    name: Topk
+    name: "Topk"
     topk: 5
     class_id_map_file: "ppcls/utils/imagenet1k_label_list.txt"
 
 Metric:
-    Train:
+  Train:
     - TopkAcc:
         topk: [1, 5]
-    Eval:
+  Eval:
     - TopkAcc:
         topk: [1, 5]

ppcls/configs/ImageNet/HRNet/HRNet_W44_C.yaml

@@ -30,10 +30,10 @@ Loss:
 
 
 Optimizer:
-  name: Momentum
+  name: "Momentum"
   momentum: 0.9
   lr:
-    name: Piecewise
+    name: "Piecewise"
     learning_rate: 0.1
     decay_epochs: [30, 60, 90]
     values: [0.1, 0.01, 0.001, 0.0001]
@@ -46,78 +46,80 @@ Optimizer:
 DataLoader:
   Train:
     dataset:
-        name: ImageNetDataset
-        image_root: "./dataset/ILSVRC2012/"
-        cls_label_path: "./dataset/ILSVRC2012/train_list.txt"
-        transform_ops:
-          - RandCropImage:
-              size: 224
-          - RandFlipImage:
-              flip_code: 1
-          - NormalizeImage:
-              scale: 0.00392157
-              mean: [0.485, 0.456, 0.406]
-              std: [0.229, 0.224, 0.225]
-              order: ''
+      name: "ImageNetDataset"
+      image_root: "./dataset/ILSVRC2012/"
+      cls_label_path: "./dataset/ILSVRC2012/train_list.txt"
+      transform_ops:
+        - RandCropImage:
+            size: 224
+        - RandFlipImage:
+            flip_code: 1
+        - NormalizeImage:
+            scale: 1.0/255.0
+            mean: [0.485, 0.456, 0.406]
+            std: [0.229, 0.224, 0.225]
+            order: ''
 
     sampler:
-        name: DistributedBatchSampler
-        batch_size: 64
-        drop_last: False
-        shuffle: True
+      name: "DistributedBatchSampler"
+      batch_size: 64
+      drop_last: False
+      shuffle: True
     loader:
-        num_workers: 6
-        use_shared_memory: True
+      num_workers: 4
+      use_shared_memory: True
 
   Eval:
     # TOTO: modify to the latest trainer
     dataset: 
-        name: ImageNetDataset
-        image_root: "./dataset/ILSVRC2012/"
-        cls_label_path: "./dataset/ILSVRC2012/val_list.txt"
-        transform_ops:
-          - ResizeImage:
-              size: 224
-          - NormalizeImage:
-              scale: 0.00392157
-              mean: [0.485, 0.456, 0.406]
-              std: [0.229, 0.224, 0.225]
-              order: ''
+      name: "ImageNetDataset"
+      image_root: "./dataset/ILSVRC2012/"
+      cls_label_path: "./dataset/ILSVRC2012/val_list.txt"
+      transform_ops:
+        - ResizeImage:
+            resize_short: 256
+        - CropImage:
+            size: 224
+        - NormalizeImage:
+            scale: 1.0/255.0
+            mean: [0.485, 0.456, 0.406]
+            std: [0.229, 0.224, 0.225]
+            order: ''
     sampler:
-        name: DistributedBatchSampler
-        batch_size: 64
-        drop_last: False
-        shuffle: False
+      name: "DistributedBatchSampler"
+      batch_size: 64
+      drop_last: False
+      shuffle: False
     loader:
-        num_workers: 6
-        use_shared_memory: True
+      num_workers: 4
+      use_shared_memory: True
 
 Infer:
   infer_imgs: "docs/images/whl/demo.jpg"
   batch_size: 10
   transforms:
-      - DecodeImage:
-          to_rgb: True
-          channel_first: False
-      - ResizeImage:
-          resize_short: 256
-      - CropImage:
-          size: 224
-      - NormalizeImage:
-          scale: 1.0/255.0
-          mean: [0.485, 0.456, 0.406]
-          std: [0.229, 0.224, 0.225]
-          order: ''
-      - ToCHWImage:
+    - DecodeImage:
+        to_rgb: True
+        channel_first: False
+    - ResizeImage:
+        resize_short: 256
+    - CropImage:
+        size: 224
+    - NormalizeImage:
+        scale: 1.0/255.0
+        mean: [0.485, 0.456, 0.406]
+        std: [0.229, 0.224, 0.225]
+        order: ''
+    - ToCHWImage:
   PostProcess:
-    name: Topk
+    name: "Topk"
     topk: 5
     class_id_map_file: "ppcls/utils/imagenet1k_label_list.txt"
 
 Metric:
-    Train:
+  Train:
     - TopkAcc:
         topk: [1, 5]
-    Eval:
+  Eval:
     - TopkAcc:
         topk: [1, 5]

ppcls/configs/ImageNet/HRNet/HRNet_W48_C.yaml

@@ -30,10 +30,10 @@ Loss:
 
 
 Optimizer:
-  name: Momentum
+  name: "Momentum"
   momentum: 0.9
   lr:
-    name: Piecewise
+    name: "Piecewise"
     learning_rate: 0.1
     decay_epochs: [30, 60, 90]
     values: [0.1, 0.01, 0.001, 0.0001]
@@ -46,80 +46,80 @@ Optimizer:
 DataLoader:
   Train:
     dataset:
-        name: ImageNetDataset
-        image_root: "./dataset/ILSVRC2012/"
-        cls_label_path: "./dataset/ILSVRC2012/train_list.txt"
-        transform_ops:
-          - RandCropImage:
-              size: 224
-          - RandFlipImage:
-              flip_code: 1
-          - NormalizeImage:
-              scale: 0.00392157
-              mean: [0.485, 0.456, 0.406]
-              std: [0.229, 0.224, 0.225]
-              order: ''
+      name: "ImageNetDataset"
+      image_root: "./dataset/ILSVRC2012/"
+      cls_label_path: "./dataset/ILSVRC2012/train_list.txt"
+      transform_ops:
+        - RandCropImage:
+            size: 224
+        - RandFlipImage:
+            flip_code: 1
+        - NormalizeImage:
+            scale: 1.0/255.0
+            mean: [0.485, 0.456, 0.406]
+            std: [0.229, 0.224, 0.225]
+            order: ''
 
     sampler:
-        name: DistributedBatchSampler
-        batch_size: 64
-        drop_last: False
-        shuffle: True
+      name: "DistributedBatchSampler"
+      batch_size: 64
+      drop_last: False
+      shuffle: True
     loader:
-        num_workers: 6
-        use_shared_memory: True
+      num_workers: 4
+      use_shared_memory: True
 
   Eval:
     # TOTO: modify to the latest trainer
     dataset: 
-        name: ImageNetDataset
-        image_root: "./dataset/ILSVRC2012/"
-        cls_label_path: "./dataset/ILSVRC2012/val_list.txt"
-        transform_ops:
-          - ResizeImage:
-              resize_short: 256
-          - CropImage:
-              size: 224
-          - NormalizeImage:
-              scale: 0.00392157
-              mean: [0.485, 0.456, 0.406]
-              std: [0.229, 0.224, 0.225]
-              order: ''
+      name: "ImageNetDataset"
+      image_root: "./dataset/ILSVRC2012/"
+      cls_label_path: "./dataset/ILSVRC2012/val_list.txt"
+      transform_ops:
+        - ResizeImage:
+            resize_short: 256
+        - CropImage:
+            size: 224
+        - NormalizeImage:
+            scale: 1.0/255.0
+            mean: [0.485, 0.456, 0.406]
+            std: [0.229, 0.224, 0.225]
+            order: ''
     sampler:
-        name: DistributedBatchSampler
-        batch_size: 64
-        drop_last: False
-        shuffle: False
+      name: "DistributedBatchSampler"
+      batch_size: 64
+      drop_last: False
+      shuffle: False
     loader:
-        num_workers: 6
-        use_shared_memory: True
+      num_workers: 4
+      use_shared_memory: True
 
 Infer:
   infer_imgs: "docs/images/whl/demo.jpg"
   batch_size: 10
   transforms:
-      - DecodeImage:
-          to_rgb: True
-          channel_first: False
-      - ResizeImage:
-          resize_short: 256
-      - CropImage:
-          size: 224
-      - NormalizeImage:
-          scale: 1.0/255.0
-          mean: [0.485, 0.456, 0.406]
-          std: [0.229, 0.224, 0.225]
-          order: ''
-      - ToCHWImage:
+    - DecodeImage:
+        to_rgb: True
+        channel_first: False
+    - ResizeImage:
+        resize_short: 256
+    - CropImage:
+        size: 224
+    - NormalizeImage:
+        scale: 1.0/255.0
+        mean: [0.485, 0.456, 0.406]
+        std: [0.229, 0.224, 0.225]
+        order: ''
+    - ToCHWImage:
   PostProcess:
-    name: Topk
+    name: "Topk"
     topk: 5
     class_id_map_file: "ppcls/utils/imagenet1k_label_list.txt"
 
 Metric:
-    Train:
+  Train:
     - TopkAcc:
         topk: [1, 5]
-    Eval:
+  Eval:
     - TopkAcc:
         topk: [1, 5]

ppcls/configs/ImageNet/HRNet/HRNet_W64_C.yaml

@@ -30,10 +30,10 @@ Loss:
 
 
 Optimizer:
-  name: Momentum
+  name: "Momentum"
   momentum: 0.9
   lr:
-    name: Piecewise
+    name: "Piecewise"
     learning_rate: 0.1
     decay_epochs: [30, 60, 90]
     values: [0.1, 0.01, 0.001, 0.0001]
@@ -46,80 +46,80 @@ Optimizer:
 DataLoader:
   Train:
     dataset:
-        name: ImageNetDataset
-        image_root: "./dataset/ILSVRC2012/"
-        cls_label_path: "./dataset/ILSVRC2012/train_list.txt"
-        transform_ops:
-          - RandCropImage:
-              size: 224
-          - RandFlipImage:
-              flip_code: 1
-          - NormalizeImage:
-              scale: 0.00392157
-              mean: [0.485, 0.456, 0.406]
-              std: [0.229, 0.224, 0.225]
-              order: ''
+      name: "ImageNetDataset"
+      image_root: "./dataset/ILSVRC2012/"
+      cls_label_path: "./dataset/ILSVRC2012/train_list.txt"
+      transform_ops:
+        - RandCropImage:
+            size: 224
+        - RandFlipImage:
+            flip_code: 1
+        - NormalizeImage:
+            scale: 1.0/255.0
+            mean: [0.485, 0.456, 0.406]
+            std: [0.229, 0.224, 0.225]
+            order: ''
 
     sampler:
-        name: DistributedBatchSampler
-        batch_size: 64
-        drop_last: False
-        shuffle: True
+      name: "DistributedBatchSampler"
+      batch_size: 64
+      drop_last: False
+      shuffle: True
     loader:
-        num_workers: 6
-        use_shared_memory: True
+      num_workers: 4
+      use_shared_memory: True
 
   Eval:
     # TOTO: modify to the latest trainer
     dataset: 
-        name: ImageNetDataset
-        image_root: "./dataset/ILSVRC2012/"
-        cls_label_path: "./dataset/ILSVRC2012/val_list.txt"
-        transform_ops:
-          - ResizeImage:
-              resize_short: 256
-          - CropImage:
-              size: 224
-          - NormalizeImage:
-              scale: 0.00392157
-              mean: [0.485, 0.456, 0.406]
-              std: [0.229, 0.224, 0.225]
-              order: ''
+      name: "ImageNetDataset"
+      image_root: "./dataset/ILSVRC2012/"
+      cls_label_path: "./dataset/ILSVRC2012/val_list.txt"
+      transform_ops:
+        - ResizeImage:
+            resize_short: 256
+        - CropImage:
+            size: 224
+        - NormalizeImage:
+            scale: 1.0/255.0
+            mean: [0.485, 0.456, 0.406]
+            std: [0.229, 0.224, 0.225]
+            order: ''
     sampler:
-        name: DistributedBatchSampler
-        batch_size: 64
-        drop_last: False
-        shuffle: False
+      name: "DistributedBatchSampler"
+      batch_size: 64
+      drop_last: False
+      shuffle: False
     loader:
-        num_workers: 6
-        use_shared_memory: True
+      num_workers: 4
+      use_shared_memory: True
 
 Infer:
   infer_imgs: "docs/images/whl/demo.jpg"
   batch_size: 10
   transforms:
-      - DecodeImage:
-          to_rgb: True
-          channel_first: False
-      - ResizeImage:
-          resize_short: 256
-      - CropImage:
-          size: 224
-      - NormalizeImage:
-          scale: 1.0/255.0
-          mean: [0.485, 0.456, 0.406]
-          std: [0.229, 0.224, 0.225]
-          order: ''
-      - ToCHWImage:
+    - DecodeImage:
+        to_rgb: True
+        channel_first: False
+    - ResizeImage:
+        resize_short: 256
+    - CropImage:
+        size: 224
+    - NormalizeImage:
+        scale: 1.0/255.0
+        mean: [0.485, 0.456, 0.406]
+        std: [0.229, 0.224, 0.225]
+        order: ''
+    - ToCHWImage:
   PostProcess:
-    name: Topk
+    name: "Topk"
     topk: 5
     class_id_map_file: "ppcls/utils/imagenet1k_label_list.txt"
 
 Metric:
-    Train:
+  Train:
     - TopkAcc:
         topk: [1, 5]
-    Eval:
+  Eval:
     - TopkAcc:
         topk: [1, 5]

ppcls/configs/ImageNet/Ineption/InceptionV3.yaml

@@ -12,7 +12,7 @@ Global:
   print_batch_step: 10
   use_visualdl: False
   # used for static mode and model export
-  image_shape: [3, 224, 224]
+  image_shape: [3, 299, 299]
   save_inference_dir: "./inference"
 
 # model architecture
@@ -24,19 +24,18 @@ Loss:
   Train:
     - CELoss:
         weight: 1.0
+        epsilon: 0.1
   Eval:
     - CELoss:
         weight: 1.0
 
 
 Optimizer:
-  name: Momentum
+  name: "Momentum"
   momentum: 0.9
   lr:
-    name: Cosine
+    name: "Cosine"
     learning_rate: 0.045
-    decay_epochs: [30, 60, 90]
-    values: [0.1, 0.01, 0.001, 0.0001]
   regularizer:
     name: 'L2'
     coeff: 0.0001
@@ -46,80 +45,80 @@ Optimizer:
 DataLoader:
   Train:
     dataset:
-        name: ImageNetDataset
-        image_root: "./dataset/ILSVRC2012/"
-        cls_label_path: "./dataset/ILSVRC2012/train_list.txt"
-        transform_ops:
-          - RandCropImage:
-              size: 299
-          - RandFlipImage:
-              flip_code: 1
-          - NormalizeImage:
-              scale: 0.00392157
-              mean: [0.485, 0.456, 0.406]
-              std: [0.229, 0.224, 0.225]
-              order: ''
+      name: "ImageNetDataset"
+      image_root: "./dataset/ILSVRC2012/"
+      cls_label_path: "./dataset/ILSVRC2012/train_list.txt"
+      transform_ops:
+        - RandCropImage:
+            size: 299
+        - RandFlipImage:
+            flip_code: 1
+        - NormalizeImage:
+            scale: 1.0/255.0
+            mean: [0.485, 0.456, 0.406]
+            std: [0.229, 0.224, 0.225]
+            order: ''
 
     sampler:
-        name: DistributedBatchSampler
-        batch_size: 64
-        drop_last: False
-        shuffle: True
+      name: "DistributedBatchSampler"
+      batch_size: 64
+      drop_last: False
+      shuffle: True
     loader:
-        num_workers: 6
-        use_shared_memory: True
+      num_workers: 4
+      use_shared_memory: True
 
   Eval:
     # TOTO: modify to the latest trainer
     dataset: 
-        name: ImageNetDataset
-        image_root: "./dataset/ILSVRC2012/"
-        cls_label_path: "./dataset/ILSVRC2012/val_list.txt"
-        transform_ops:
-          - ResizeImage:
-              resize_short: 320
-          - CropImage:
-              size: 299
-          - NormalizeImage:
-              scale: 0.00392157
-              mean: [0.485, 0.456, 0.406]
-              std: [0.229, 0.224, 0.225]
-              order: ''
+      name: "ImageNetDataset"
+      image_root: "./dataset/ILSVRC2012/"
+      cls_label_path: "./dataset/ILSVRC2012/val_list.txt"
+      transform_ops:
+        - ResizeImage:
+            resize_short: 320
+        - CropImage:
+            size: 299
+        - NormalizeImage:
+            scale: 1.0/255.0
+            mean: [0.485, 0.456, 0.406]
+            std: [0.229, 0.224, 0.225]
+            order: ''
     sampler:
-        name: DistributedBatchSampler
-        batch_size: 64
-        drop_last: False
-        shuffle: False
+      name: "DistributedBatchSampler"
+      batch_size: 64
+      drop_last: False
+      shuffle: False
     loader:
-        num_workers: 6
-        use_shared_memory: True
+      num_workers: 4
+      use_shared_memory: True
 
 Infer:
   infer_imgs: "docs/images/whl/demo.jpg"
   batch_size: 10
   transforms:
-      - DecodeImage:
-          to_rgb: True
-          channel_first: False
-      - ResizeImage:
-          resize_short: 320
-      - CropImage:
-          size: 299
-      - NormalizeImage:
-          scale: 1.0/255.0
-          mean: [0.485, 0.456, 0.406]
-          std: [0.229, 0.224, 0.225]
-          order: ''
-      - ToCHWImage:
+    - DecodeImage:
+        to_rgb: True
+        channel_first: False
+    - ResizeImage:
+        resize_short: 320
+    - CropImage:
+        size: 299
+    - NormalizeImage:
+        scale: 1.0/255.0
+        mean: [0.485, 0.456, 0.406]
+        std: [0.229, 0.224, 0.225]
+        order: ''
+    - ToCHWImage:
   PostProcess:
-    name: Topk
+    name: "Topk"
     topk: 5
     class_id_map_file: "ppcls/utils/imagenet1k_label_list.txt"
 
 Metric:
-    Train:
+  Train:
     - TopkAcc:
         topk: [1, 5]
-    Eval:
+  Eval:
     - TopkAcc:
         topk: [1, 5]

ppcls/configs/ImageNet/MobileNetV1/MobileNetV1.yaml

@@ -30,96 +30,96 @@ Loss:
 
 
 Optimizer:
-  name: Momentum
+  name: "Momentum"
   momentum: 0.9
   lr:
-    name: Piecewise
+    name: "Piecewise"
     learning_rate: 0.1
     decay_epochs: [30, 60, 90]
     values: [0.1, 0.01, 0.001, 0.0001]
   regularizer:
     name: 'L2'
-    coeff: 0.00003
+    coeff: 0.0003
 
 
 # data loader for train and eval
 DataLoader:
   Train:
     dataset:
-        name: ImageNetDataset
-        image_root: "./dataset/ILSVRC2012/"
-        cls_label_path: "./dataset/ILSVRC2012/train_list.txt"
-        transform_ops:
-          - RandCropImage:
-              size: 224
-          - RandFlipImage:
-              flip_code: 1
-          - NormalizeImage:
-              scale: 0.00392157
-              mean: [0.485, 0.456, 0.406]
-              std: [0.229, 0.224, 0.225]
-              order: ''
+      name: "ImageNetDataset"
+      image_root: "./dataset/ILSVRC2012/"
+      cls_label_path: "./dataset/ILSVRC2012/train_list.txt"
+      transform_ops:
+        - RandCropImage:
+            size: 224
+        - RandFlipImage:
+            flip_code: 1
+        - NormalizeImage:
+            scale: 1.0/255.0
+            mean: [0.485, 0.456, 0.406]
+            std: [0.229, 0.224, 0.225]
+            order: ''
 
     sampler:
-        name: DistributedBatchSampler
-        batch_size: 64
-        drop_last: False
-        shuffle: True
+      name: "DistributedBatchSampler"
+      batch_size: 64
+      drop_last: False
+      shuffle: True
     loader:
-        num_workers: 6
-        use_shared_memory: True
+      num_workers: 4
+      use_shared_memory: True
 
   Eval:
     # TOTO: modify to the latest trainer
     dataset: 
-        name: ImageNetDataset
-        image_root: "./dataset/ILSVRC2012/"
-        cls_label_path: "./dataset/ILSVRC2012/val_list.txt"
-        transform_ops:
-          - ResizeImage:
-              resize_short: 256
-          - CropImage:
-              size: 224
-          - NormalizeImage:
-              scale: 0.00392157
-              mean: [0.485, 0.456, 0.406]
-              std: [0.229, 0.224, 0.225]
-              order: ''
+      name: "ImageNetDataset"
+      image_root: "./dataset/ILSVRC2012/"
+      cls_label_path: "./dataset/ILSVRC2012/val_list.txt"
+      transform_ops:
+        - ResizeImage:
+            resize_short: 256
+        - CropImage:
+            size: 224
+        - NormalizeImage:
+            scale: 1.0/255.0
+            mean: [0.485, 0.456, 0.406]
+            std: [0.229, 0.224, 0.225]
+            order: ''
     sampler:
-        name: DistributedBatchSampler
-        batch_size: 64
-        drop_last: False
-        shuffle: False
+      name: "DistributedBatchSampler"
+      batch_size: 64
+      drop_last: False
+      shuffle: False
     loader:
-        num_workers: 6
-        use_shared_memory: True
+      num_workers: 4
+      use_shared_memory: True
 
 Infer:
   infer_imgs: "docs/images/whl/demo.jpg"
   batch_size: 10
   transforms:
-      - DecodeImage:
-          to_rgb: True
-          channel_first: False
-      - ResizeImage:
-          resize_short: 256
-      - CropImage:
-          size: 224
-      - NormalizeImage:
-          scale: 1.0/255.0
-          mean: [0.485, 0.456, 0.406]
-          std: [0.229, 0.224, 0.225]
-          order: ''
-      - ToCHWImage:
+    - DecodeImage:
+        to_rgb: True
+        channel_first: False
+    - ResizeImage:
+        resize_short: 256
+    - CropImage:
+        size: 224
+    - NormalizeImage:
+        scale: 1.0/255.0
+        mean: [0.485, 0.456, 0.406]
+        std: [0.229, 0.224, 0.225]
+        order: ''
+    - ToCHWImage:
   PostProcess:
-    name: Topk
+    name: "Topk"
     topk: 5
     class_id_map_file: "ppcls/utils/imagenet1k_label_list.txt"
 
 Metric:
-    Train:
+  Train:
     - TopkAcc:
         topk: [1, 5]
-    Eval:
+  Eval:
     - TopkAcc:
         topk: [1, 5]

ppcls/configs/ImageNet/MobileNetV1/MobileNetV1_x0_25.yaml

@@ -30,96 +30,96 @@ Loss:
 
 
 Optimizer:
-  name: Momentum
+  name: "Momentum"
   momentum: 0.9
   lr:
-    name: Piecewise
+    name: "Piecewise"
     learning_rate: 0.1
     decay_epochs: [30, 60, 90]
     values: [0.1, 0.01, 0.001, 0.0001]
   regularizer:
     name: 'L2'
-    coeff: 0.00003
+    coeff: 0.0003
 
 
 # data loader for train and eval
 DataLoader:
   Train:
     dataset:
-        name: ImageNetDataset
-        image_root: "./dataset/ILSVRC2012/"
-        cls_label_path: "./dataset/ILSVRC2012/train_list.txt"
-        transform_ops:
-          - RandCropImage:
-              size: 224
-          - RandFlipImage:
-              flip_code: 1
-          - NormalizeImage:
-              scale: 0.00392157
-              mean: [0.485, 0.456, 0.406]
-              std: [0.229, 0.224, 0.225]
-              order: ''
+      name: "ImageNetDataset"
+      image_root: "./dataset/ILSVRC2012/"
+      cls_label_path: "./dataset/ILSVRC2012/train_list.txt"
+      transform_ops:
+        - RandCropImage:
+            size: 224
+        - RandFlipImage:
+            flip_code: 1
+        - NormalizeImage:
+            scale: 1.0/255.0
+            mean: [0.485, 0.456, 0.406]
+            std: [0.229, 0.224, 0.225]
+            order: ''
 
     sampler:
-        name: DistributedBatchSampler
-        batch_size: 64
-        drop_last: False
-        shuffle: True
+      name: "DistributedBatchSampler"
+      batch_size: 64
+      drop_last: False
+      shuffle: True
     loader:
-        num_workers: 6
-        use_shared_memory: True
+      num_workers: 4
+      use_shared_memory: True
 
   Eval:
     # TOTO: modify to the latest trainer
     dataset: 
-        name: ImageNetDataset
-        image_root: "./dataset/ILSVRC2012/"
-        cls_label_path: "./dataset/ILSVRC2012/val_list.txt"
-        transform_ops:
-          - ResizeImage:
-              resize_short: 256
-          - CropImage:
-              size: 224
-          - NormalizeImage:
-              scale: 0.00392157
-              mean: [0.485, 0.456, 0.406]
-              std: [0.229, 0.224, 0.225]
-              order: ''
+      name: "ImageNetDataset"
+      image_root: "./dataset/ILSVRC2012/"
+      cls_label_path: "./dataset/ILSVRC2012/val_list.txt"
+      transform_ops:
+        - ResizeImage:
+            resize_short: 256
+        - CropImage:
+            size: 224
+        - NormalizeImage:
+            scale: 1.0/255.0
+            mean: [0.485, 0.456, 0.406]
+            std: [0.229, 0.224, 0.225]
+            order: ''
     sampler:
-        name: DistributedBatchSampler
-        batch_size: 64
-        drop_last: False
-        shuffle: False
+      name: "DistributedBatchSampler"
+      batch_size: 64
+      drop_last: False
+      shuffle: False
     loader:
-        num_workers: 6
-        use_shared_memory: True
+      num_workers: 4
+      use_shared_memory: True
 
 Infer:
   infer_imgs: "docs/images/whl/demo.jpg"
   batch_size: 10
   transforms:
-      - DecodeImage:
-          to_rgb: True
-          channel_first: False
-      - ResizeImage:
-          resize_short: 256
-      - CropImage:
-          size: 224
-      - NormalizeImage:
-          scale: 1.0/255.0
-          mean: [0.485, 0.456, 0.406]
-          std: [0.229, 0.224, 0.225]
-          order: ''
-      - ToCHWImage:
+    - DecodeImage:
+        to_rgb: True
+        channel_first: False
+    - ResizeImage:
+        resize_short: 256
+    - CropImage:
+        size: 224
+    - NormalizeImage:
+        scale: 1.0/255.0
+        mean: [0.485, 0.456, 0.406]
+        std: [0.229, 0.224, 0.225]
+        order: ''
+    - ToCHWImage:
   PostProcess:
-    name: Topk
+    name: "Topk"
     topk: 5
     class_id_map_file: "ppcls/utils/imagenet1k_label_list.txt"
 
 Metric:
-    Train:
+  Train:
     - TopkAcc:
         topk: [1, 5]
-    Eval:
+  Eval:
     - TopkAcc:
         topk: [1, 5]

ppcls/configs/ImageNet/MobileNetV1/MobileNetV1_x0_5.yaml

@@ -30,96 +30,96 @@ Loss:
 
 
 Optimizer:
-  name: Momentum
+  name: "Momentum"
   momentum: 0.9
   lr:
-    name: Piecewise
+    name: "Piecewise"
     learning_rate: 0.1
     decay_epochs: [30, 60, 90]
     values: [0.1, 0.01, 0.001, 0.0001]
   regularizer:
     name: 'L2'
-    coeff: 0.00003
+    coeff: 0.0003
 
 
 # data loader for train and eval
 DataLoader:
   Train:
     dataset:
-        name: ImageNetDataset
-        image_root: "./dataset/ILSVRC2012/"
-        cls_label_path: "./dataset/ILSVRC2012/train_list.txt"
-        transform_ops:
-          - RandCropImage:
-              size: 224
-          - RandFlipImage:
-              flip_code: 1
-          - NormalizeImage:
-              scale: 0.00392157
-              mean: [0.485, 0.456, 0.406]
-              std: [0.229, 0.224, 0.225]
-              order: ''
+      name: "ImageNetDataset"
+      image_root: "./dataset/ILSVRC2012/"
+      cls_label_path: "./dataset/ILSVRC2012/train_list.txt"
+      transform_ops:
+        - RandCropImage:
+            size: 224
+        - RandFlipImage:
+            flip_code: 1
+        - NormalizeImage:
+            scale: 1.0/255.0
+            mean: [0.485, 0.456, 0.406]
+            std: [0.229, 0.224, 0.225]
+            order: ''
 
     sampler:
-        name: DistributedBatchSampler
-        batch_size: 64
-        drop_last: False
-        shuffle: True
+      name: "DistributedBatchSampler"
+      batch_size: 64
+      drop_last: False
+      shuffle: True
     loader:
-        num_workers: 6
-        use_shared_memory: True
+      num_workers: 4
+      use_shared_memory: True
 
   Eval:
     # TOTO: modify to the latest trainer
     dataset: 
-        name: ImageNetDataset
-        image_root: "./dataset/ILSVRC2012/"
-        cls_label_path: "./dataset/ILSVRC2012/val_list.txt"
-        transform_ops:
-          - ResizeImage:
-              resize_short: 256
-          - CropImage:
-              size: 224
-          - NormalizeImage:
-              scale: 0.00392157
-              mean: [0.485, 0.456, 0.406]
-              std: [0.229, 0.224, 0.225]
-              order: ''
+      name: "ImageNetDataset"
+      image_root: "./dataset/ILSVRC2012/"
+      cls_label_path: "./dataset/ILSVRC2012/val_list.txt"
+      transform_ops:
+        - ResizeImage:
+            resize_short: 256
+        - CropImage:
+            size: 224
+        - NormalizeImage:
+            scale: 1.0/255.0
+            mean: [0.485, 0.456, 0.406]
+            std: [0.229, 0.224, 0.225]
+            order: ''
     sampler:
-        name: DistributedBatchSampler
-        batch_size: 64
-        drop_last: False
-        shuffle: False
+      name: "DistributedBatchSampler"
+      batch_size: 64
+      drop_last: False
+      shuffle: False
     loader:
-        num_workers: 6
-        use_shared_memory: True
+      num_workers: 4
+      use_shared_memory: True
 
 Infer:
   infer_imgs: "docs/images/whl/demo.jpg"
   batch_size: 10
   transforms:
-      - DecodeImage:
-          to_rgb: True
-          channel_first: False
-      - ResizeImage:
-          resize_short: 256
-      - CropImage:
-          size: 224
-      - NormalizeImage:
-          scale: 1.0/255.0
-          mean: [0.485, 0.456, 0.406]
-          std: [0.229, 0.224, 0.225]
-          order: ''
-      - ToCHWImage:
+    - DecodeImage:
+        to_rgb: True
+        channel_first: False
+    - ResizeImage:
+        resize_short: 256
+    - CropImage:
+        size: 224
+    - NormalizeImage:
+        scale: 1.0/255.0
+        mean: [0.485, 0.456, 0.406]
+        std: [0.229, 0.224, 0.225]
+        order: ''
+    - ToCHWImage:
   PostProcess:
-    name: Topk
+    name: "Topk"
     topk: 5
     class_id_map_file: "ppcls/utils/imagenet1k_label_list.txt"
 
 Metric:
-    Train:
+  Train:
     - TopkAcc:
         topk: [1, 5]
-    Eval:
+  Eval:
     - TopkAcc:
         topk: [1, 5]

ppcls/configs/ImageNet/MobileNetV1/MobileNetV1_x0_75.yaml

@@ -30,96 +30,96 @@ Loss:
 
 
 Optimizer:
-  name: Momentum
+  name: "Momentum"
   momentum: 0.9
   lr:
-    name: Piecewise
+    name: "Piecewise"
     learning_rate: 0.1
     decay_epochs: [30, 60, 90]
     values: [0.1, 0.01, 0.001, 0.0001]
   regularizer:
     name: 'L2'
-    coeff: 0.00003
+    coeff: 0.0003
 
 
 # data loader for train and eval
 DataLoader:
   Train:
     dataset:
-        name: ImageNetDataset
-        image_root: "./dataset/ILSVRC2012/"
-        cls_label_path: "./dataset/ILSVRC2012/train_list.txt"
-        transform_ops:
-          - RandCropImage:
-              size: 224
-          - RandFlipImage:
-              flip_code: 1
-          - NormalizeImage:
-              scale: 0.00392157
-              mean: [0.485, 0.456, 0.406]
-              std: [0.229, 0.224, 0.225]
-              order: ''
+      name: "ImageNetDataset"
+      image_root: "./dataset/ILSVRC2012/"
+      cls_label_path: "./dataset/ILSVRC2012/train_list.txt"
+      transform_ops:
+        - RandCropImage:
+            size: 224
+        - RandFlipImage:
+            flip_code: 1
+        - NormalizeImage:
+            scale: 1.0/255.0
+            mean: [0.485, 0.456, 0.406]
+            std: [0.229, 0.224, 0.225]
+            order: ''
 
     sampler:
-        name: DistributedBatchSampler
-        batch_size: 64
-        drop_last: False
-        shuffle: True
+      name: "DistributedBatchSampler"
+      batch_size: 64
+      drop_last: False
+      shuffle: True
     loader:
-        num_workers: 6
-        use_shared_memory: True
+      num_workers: 4
+      use_shared_memory: True
 
   Eval:
     # TOTO: modify to the latest trainer
     dataset: 
-        name: ImageNetDataset
-        image_root: "./dataset/ILSVRC2012/"
-        cls_label_path: "./dataset/ILSVRC2012/val_list.txt"
-        transform_ops:
-          - ResizeImage:
-              resize_short: 256
-          - CropImage:
-              size: 224
-          - NormalizeImage:
-              scale: 0.00392157
-              mean: [0.485, 0.456, 0.406]
-              std: [0.229, 0.224, 0.225]
-              order: ''
+      name: "ImageNetDataset"
+      image_root: "./dataset/ILSVRC2012/"
+      cls_label_path: "./dataset/ILSVRC2012/val_list.txt"
+      transform_ops:
+        - ResizeImage:
+            resize_short: 256
+        - CropImage:
+            size: 224
+        - NormalizeImage:
+            scale: 1.0/255.0
+            mean: [0.485, 0.456, 0.406]
+            std: [0.229, 0.224, 0.225]
+            order: ''
     sampler:
-        name: DistributedBatchSampler
-        batch_size: 64
-        drop_last: False
-        shuffle: False
+      name: "DistributedBatchSampler"
+      batch_size: 64
+      drop_last: False
+      shuffle: False
     loader:
-        num_workers: 6
-        use_shared_memory: True
+      num_workers: 4
+      use_shared_memory: True
 
 Infer:
   infer_imgs: "docs/images/whl/demo.jpg"
   batch_size: 10
   transforms:
-      - DecodeImage:
-          to_rgb: True
-          channel_first: False
-      - ResizeImage:
-          resize_short: 256
-      - CropImage:
-          size: 224
-      - NormalizeImage:
-          scale: 1.0/255.0
-          mean: [0.485, 0.456, 0.406]
-          std: [0.229, 0.224, 0.225]
-          order: ''
-      - ToCHWImage:
+    - DecodeImage:
+        to_rgb: True
+        channel_first: False
+    - ResizeImage:
+        resize_short: 256
+    - CropImage:
+        size: 224
+    - NormalizeImage:
+        scale: 1.0/255.0
+        mean: [0.485, 0.456, 0.406]
+        std: [0.229, 0.224, 0.225]
+        order: ''
+    - ToCHWImage:
   PostProcess:
-    name: Topk
+    name: "Topk"
     topk: 5
     class_id_map_file: "ppcls/utils/imagenet1k_label_list.txt"
 
 Metric:
-    Train:
+  Train:
     - TopkAcc:
         topk: [1, 5]
-    Eval:
+  Eval:
     - TopkAcc:
         topk: [1, 5]

ppcls/configs/ImageNet/MobileNetV3/MobileNetV3_large_x0_35.yaml

@@ -31,10 +31,10 @@ Loss:
 
 
 Optimizer:
-  name: Momentum
+  name: "Momentum"
   momentum: 0.9
   lr:
-    name: Cosine
+    name: "Cosine"
     learning_rate: 1.3
   regularizer:
     name: 'L2'
@@ -45,80 +45,80 @@ Optimizer:
 DataLoader:
   Train:
     dataset:
-        name: ImageNetDataset
-        image_root: "./dataset/ILSVRC2012/"
-        cls_label_path: "./dataset/ILSVRC2012/train_list.txt"
-        transform_ops:
-          - RandCropImage:
-              size: 224
-          - RandFlipImage:
-              flip_code: 1
-          - NormalizeImage:
-              scale: 0.00392157
-              mean: [0.485, 0.456, 0.406]
-              std: [0.229, 0.224, 0.225]
-              order: ''
+      name: "ImageNetDataset"
+      image_root: "./dataset/ILSVRC2012/"
+      cls_label_path: "./dataset/ILSVRC2012/train_list.txt"
+      transform_ops:
+        - RandCropImage:
+            size: 224
+        - RandFlipImage:
+            flip_code: 1
+        - NormalizeImage:
+            scale: 1.0/255.0
+            mean: [0.485, 0.456, 0.406]
+            std: [0.229, 0.224, 0.225]
+            order: ''
 
     sampler:
-        name: DistributedBatchSampler
-        batch_size: 512
-        drop_last: False
-        shuffle: True
+      name: "DistributedBatchSampler"
+      batch_size: 512
+      drop_last: False
+      shuffle: True
     loader:
-        num_workers: 6
-        use_shared_memory: True
+      num_workers: 4
+      use_shared_memory: True
 
   Eval:
     # TOTO: modify to the latest trainer
     dataset: 
-        name: ImageNetDataset
-        image_root: "./dataset/ILSVRC2012/"
-        cls_label_path: "./dataset/ILSVRC2012/val_list.txt"
-        transform_ops:
-          - ResizeImage:
-              resize_short: 256
-          - CropImage:
-              size: 224
-          - NormalizeImage:
-              scale: 0.00392157
-              mean: [0.485, 0.456, 0.406]
-              std: [0.229, 0.224, 0.225]
-              order: ''
+      name: "ImageNetDataset"
+      image_root: "./dataset/ILSVRC2012/"
+      cls_label_path: "./dataset/ILSVRC2012/val_list.txt"
+      transform_ops:
+        - ResizeImage:
+            resize_short: 256
+        - CropImage:
+            size: 224
+        - NormalizeImage:
+            scale: 1.0/255.0
+            mean: [0.485, 0.456, 0.406]
+            std: [0.229, 0.224, 0.225]
+            order: ''
     sampler:
-        name: DistributedBatchSampler
-        batch_size: 64
-        drop_last: False
-        shuffle: False
+      name: "DistributedBatchSampler"
+      batch_size: 64
+      drop_last: False
+      shuffle: False
     loader:
-        num_workers: 6
-        use_shared_memory: True
+      num_workers: 4
+      use_shared_memory: True
 
 Infer:
   infer_imgs: "docs/images/whl/demo.jpg"
   batch_size: 10
   transforms:
-      - DecodeImage:
-          to_rgb: True
-          channel_first: False
-      - ResizeImage:
-          resize_short: 256
-      - CropImage:
-          size: 224
-      - NormalizeImage:
-          scale: 1.0/255.0
-          mean: [0.485, 0.456, 0.406]
-          std: [0.229, 0.224, 0.225]
-          order: ''
-      - ToCHWImage:
+    - DecodeImage:
+        to_rgb: True
+        channel_first: False
+    - ResizeImage:
+        resize_short: 256
+    - CropImage:
+        size: 224
+    - NormalizeImage:
+        scale: 1.0/255.0
+        mean: [0.485, 0.456, 0.406]
+        std: [0.229, 0.224, 0.225]
+        order: ''
+    - ToCHWImage:
   PostProcess:
-    name: Topk
+    name: "Topk"
     topk: 5
     class_id_map_file: "ppcls/utils/imagenet1k_label_list.txt"
 
 Metric:
-    Train:
+  Train:
     - TopkAcc:
         topk: [1, 5]
-    Eval:
+  Eval:
     - TopkAcc:
         topk: [1, 5]

ppcls/configs/ImageNet/MobileNetV3/MobileNetV3_large_x0_5.yaml

@@ -31,10 +31,10 @@ Loss:
 
 
 Optimizer:
-  name: Momentum
+  name: "Momentum"
   momentum: 0.9
   lr:
-    name: Cosine
+    name: "Cosine"
     learning_rate: 1.3
   regularizer:
     name: 'L2'
@@ -45,80 +45,80 @@ Optimizer:
 DataLoader:
   Train:
     dataset:
-        name: ImageNetDataset
-        image_root: "./dataset/ILSVRC2012/"
-        cls_label_path: "./dataset/ILSVRC2012/train_list.txt"
-        transform_ops:
-          - RandCropImage:
-              size: 224
-          - RandFlipImage:
-              flip_code: 1
-          - NormalizeImage:
-              scale: 0.00392157
-              mean: [0.485, 0.456, 0.406]
-              std: [0.229, 0.224, 0.225]
-              order: ''
+      name: "ImageNetDataset"
+      image_root: "./dataset/ILSVRC2012/"
+      cls_label_path: "./dataset/ILSVRC2012/train_list.txt"
+      transform_ops:
+        - RandCropImage:
+            size: 224
+        - RandFlipImage:
+            flip_code: 1
+        - NormalizeImage:
+            scale: 1.0/255.0
+            mean: [0.485, 0.456, 0.406]
+            std: [0.229, 0.224, 0.225]
+            order: ''
 
     sampler:
-        name: DistributedBatchSampler
-        batch_size: 512
-        drop_last: False
-        shuffle: True
+      name: "DistributedBatchSampler"
+      batch_size: 512
+      drop_last: False
+      shuffle: True
     loader:
-        num_workers: 6
-        use_shared_memory: True
+      num_workers: 4
+      use_shared_memory: True
 
   Eval:
     # TOTO: modify to the latest trainer
     dataset: 
-        name: ImageNetDataset
-        image_root: "./dataset/ILSVRC2012/"
-        cls_label_path: "./dataset/ILSVRC2012/val_list.txt"
-        transform_ops:
-          - ResizeImage:
-              resize_short: 256
-          - CropImage:
-              size: 224
-          - NormalizeImage:
-              scale: 0.00392157
-              mean: [0.485, 0.456, 0.406]
-              std: [0.229, 0.224, 0.225]
-              order: ''
+      name: "ImageNetDataset"
+      image_root: "./dataset/ILSVRC2012/"
+      cls_label_path: "./dataset/ILSVRC2012/val_list.txt"
+      transform_ops:
+        - ResizeImage:
+            resize_short: 256
+        - CropImage:
+            size: 224
+        - NormalizeImage:
+            scale: 1.0/255.0
+            mean: [0.485, 0.456, 0.406]
+            std: [0.229, 0.224, 0.225]
+            order: ''
     sampler:
-        name: DistributedBatchSampler
-        batch_size: 64
-        drop_last: False
-        shuffle: False
+      name: "DistributedBatchSampler"
+      batch_size: 64
+      drop_last: False
+      shuffle: False
     loader:
-        num_workers: 6
-        use_shared_memory: True
+      num_workers: 4
+      use_shared_memory: True
 
 Infer:
   infer_imgs: "docs/images/whl/demo.jpg"
   batch_size: 10
   transforms:
-      - DecodeImage:
-          to_rgb: True
-          channel_first: False
-      - ResizeImage:
-          resize_short: 256
-      - CropImage:
-          size: 224
-      - NormalizeImage:
-          scale: 1.0/255.0
-          mean: [0.485, 0.456, 0.406]
-          std: [0.229, 0.224, 0.225]
-          order: ''
-      - ToCHWImage:
+    - DecodeImage:
+        to_rgb: True
+        channel_first: False
+    - ResizeImage:
+        resize_short: 256
+    - CropImage:
+        size: 224
+    - NormalizeImage:
+        scale: 1.0/255.0
+        mean: [0.485, 0.456, 0.406]
+        std: [0.229, 0.224, 0.225]
+        order: ''
+    - ToCHWImage:
   PostProcess:
-    name: Topk
+    name: "Topk"
     topk: 5
     class_id_map_file: "ppcls/utils/imagenet1k_label_list.txt"
 
 Metric:
-    Train:
+  Train:
     - TopkAcc:
         topk: [1, 5]
-    Eval:
+  Eval:
     - TopkAcc:
         topk: [1, 5]

ppcls/configs/ImageNet/MobileNetV3/MobileNetV3_large_x0_75.yaml

@@ -31,10 +31,10 @@ Loss:
 
 
 Optimizer:
-  name: Momentum
+  name: "Momentum"
   momentum: 0.9
   lr:
-    name: Cosine
+    name: "Cosine"
     learning_rate: 1.3
   regularizer:
     name: 'L2'
@@ -45,80 +45,80 @@ Optimizer:
 DataLoader:
   Train:
     dataset:
-        name: ImageNetDataset
-        image_root: "./dataset/ILSVRC2012/"
-        cls_label_path: "./dataset/ILSVRC2012/train_list.txt"
-        transform_ops:
-          - RandCropImage:
-              size: 224
-          - RandFlipImage:
-              flip_code: 1
-          - NormalizeImage:
-              scale: 0.00392157
-              mean: [0.485, 0.456, 0.406]
-              std: [0.229, 0.224, 0.225]
-              order: ''
+      name: "ImageNetDataset"
+      image_root: "./dataset/ILSVRC2012/"
+      cls_label_path: "./dataset/ILSVRC2012/train_list.txt"
+      transform_ops:
+        - RandCropImage:
+            size: 224
+        - RandFlipImage:
+            flip_code: 1
+        - NormalizeImage:
+            scale: 1.0/255.0
+            mean: [0.485, 0.456, 0.406]
+            std: [0.229, 0.224, 0.225]
+            order: ''
 
     sampler:
-        name: DistributedBatchSampler
-        batch_size: 512
-        drop_last: False
-        shuffle: True
+      name: "DistributedBatchSampler"
+      batch_size: 512
+      drop_last: False
+      shuffle: True
     loader:
-        num_workers: 6
-        use_shared_memory: True
+      num_workers: 4
+      use_shared_memory: True
 
   Eval:
     # TOTO: modify to the latest trainer
     dataset: 
-        name: ImageNetDataset
-        image_root: "./dataset/ILSVRC2012/"
-        cls_label_path: "./dataset/ILSVRC2012/val_list.txt"
-        transform_ops:
-          - ResizeImage:
-              resize_short: 256
-          - CropImage:
-              size: 224
-          - NormalizeImage:
-              scale: 0.00392157
-              mean: [0.485, 0.456, 0.406]
-              std: [0.229, 0.224, 0.225]
-              order: ''
+      name: "ImageNetDataset"
+      image_root: "./dataset/ILSVRC2012/"
+      cls_label_path: "./dataset/ILSVRC2012/val_list.txt"
+      transform_ops:
+        - ResizeImage:
+            resize_short: 256
+        - CropImage:
+            size: 224
+        - NormalizeImage:
+            scale: 1.0/255.0
+            mean: [0.485, 0.456, 0.406]
+            std: [0.229, 0.224, 0.225]
+            order: ''
     sampler:
-        name: DistributedBatchSampler
-        batch_size: 64
-        drop_last: False
-        shuffle: False
+      name: "DistributedBatchSampler"
+      batch_size: 64
+      drop_last: False
+      shuffle: False
     loader:
-        num_workers: 6
-        use_shared_memory: True
+      num_workers: 4
+      use_shared_memory: True
 
 Infer:
   infer_imgs: "docs/images/whl/demo.jpg"
   batch_size: 10
   transforms:
-      - DecodeImage:
-          to_rgb: True
-          channel_first: False
-      - ResizeImage:
-          resize_short: 256
-      - CropImage:
-          size: 224
-      - NormalizeImage:
-          scale: 1.0/255.0
-          mean: [0.485, 0.456, 0.406]
-          std: [0.229, 0.224, 0.225]
-          order: ''
-      - ToCHWImage:
+    - DecodeImage:
+        to_rgb: True
+        channel_first: False
+    - ResizeImage:
+        resize_short: 256
+    - CropImage:
+        size: 224
+    - NormalizeImage:
+        scale: 1.0/255.0
+        mean: [0.485, 0.456, 0.406]
+        std: [0.229, 0.224, 0.225]
+        order: ''
+    - ToCHWImage:
   PostProcess:
-    name: Topk
+    name: "Topk"
     topk: 5
     class_id_map_file: "ppcls/utils/imagenet1k_label_list.txt"
 
 Metric:
-    Train:
+  Train:
     - TopkAcc:
         topk: [1, 5]
-    Eval:
+  Eval:
     - TopkAcc:
         topk: [1, 5]

ppcls/configs/ImageNet/MobileNetV3/MobileNetV3_large_x1_0.yaml

@@ -31,10 +31,10 @@ Loss:
 
 
 Optimizer:
-  name: Momentum
+  name: "Momentum"
   momentum: 0.9
   lr:
-    name: Cosine
+    name: "Cosine"
     learning_rate: 1.3
   regularizer:
     name: 'L2'
@@ -45,81 +45,81 @@ Optimizer:
 DataLoader:
   Train:
     dataset:
-        name: ImageNetDataset
-        image_root: "./dataset/ILSVRC2012/"
-        cls_label_path: "./dataset/ILSVRC2012/train_list.txt"
-        transform_ops:
-          - RandCropImage:
-              size: 224
-          - RandFlipImage:
-              flip_code: 1
-          - AutoAugment:
-          - NormalizeImage:
-              scale: 0.00392157
-              mean: [0.485, 0.456, 0.406]
-              std: [0.229, 0.224, 0.225]
-              order: ''
+      name: "ImageNetDataset"
+      image_root: "./dataset/ILSVRC2012/"
+      cls_label_path: "./dataset/ILSVRC2012/train_list.txt"
+      transform_ops:
+        - RandCropImage:
+            size: 224
+        - RandFlipImage:
+            flip_code: 1
+        - AutoAugment:
+        - NormalizeImage:
+            scale: 1.0/255.0
+            mean: [0.485, 0.456, 0.406]
+            std: [0.229, 0.224, 0.225]
+            order: ''
 
     sampler:
-        name: DistributedBatchSampler
-        batch_size: 512
-        drop_last: False
-        shuffle: True
+      name: "DistributedBatchSampler"
+      batch_size: 512
+      drop_last: False
+      shuffle: True
     loader:
-        num_workers: 6
-        use_shared_memory: True
+      num_workers: 4
+      use_shared_memory: True
 
   Eval:
     # TOTO: modify to the latest trainer
     dataset: 
-        name: ImageNetDataset
-        image_root: "./dataset/ILSVRC2012/"
-        cls_label_path: "./dataset/ILSVRC2012/val_list.txt"
-        transform_ops:
-          - ResizeImage:
-              resize_short: 256
-          - CropImage:
-              size: 224
-          - NormalizeImage:
-              scale: 0.00392157
-              mean: [0.485, 0.456, 0.406]
-              std: [0.229, 0.224, 0.225]
-              order: ''
+      name: "ImageNetDataset"
+      image_root: "./dataset/ILSVRC2012/"
+      cls_label_path: "./dataset/ILSVRC2012/val_list.txt"
+      transform_ops:
+        - ResizeImage:
+            resize_short: 256
+        - CropImage:
+            size: 224
+        - NormalizeImage:
+            scale: 1.0/255.0
+            mean: [0.485, 0.456, 0.406]
+            std: [0.229, 0.224, 0.225]
+            order: ''
     sampler:
-        name: DistributedBatchSampler
-        batch_size: 64
-        drop_last: False
-        shuffle: False
+      name: "DistributedBatchSampler"
+      batch_size: 64
+      drop_last: False
+      shuffle: False
     loader:
-        num_workers: 6
-        use_shared_memory: True
+      num_workers: 4
+      use_shared_memory: True
 
 Infer:
   infer_imgs: "docs/images/whl/demo.jpg"
   batch_size: 10
   transforms:
-      - DecodeImage:
-          to_rgb: True
-          channel_first: False
-      - ResizeImage:
-          resize_short: 256
-      - CropImage:
-          size: 224
-      - NormalizeImage:
-          scale: 1.0/255.0
-          mean: [0.485, 0.456, 0.406]
-          std: [0.229, 0.224, 0.225]
-          order: ''
-      - ToCHWImage:
+    - DecodeImage:
+        to_rgb: True
+        channel_first: False
+    - ResizeImage:
+        resize_short: 256
+    - CropImage:
+        size: 224
+    - NormalizeImage:
+        scale: 1.0/255.0
+        mean: [0.485, 0.456, 0.406]
+        std: [0.229, 0.224, 0.225]
+        order: ''
+    - ToCHWImage:
   PostProcess:
-    name: Topk
+    name: "Topk"
     topk: 5
     class_id_map_file: "ppcls/utils/imagenet1k_label_list.txt"
 
 Metric:
-    Train:
+  Train:
     - TopkAcc:
         topk: [1, 5]
-    Eval:
+  Eval:
     - TopkAcc:
         topk: [1, 5]

ppcls/configs/ImageNet/MobileNetV3/MobileNetV3_large_x1_25.yaml

@@ -31,94 +31,94 @@ Loss:
 
 
 Optimizer:
-  name: Momentum
+  name: "Momentum"
   momentum: 0.9
   lr:
-    name: Cosine
+    name: "Cosine"
     learning_rate: 1.3
   regularizer:
     name: 'L2'
-    coeff: 0.00002
+    coeff: 0.00004
 
 
 # data loader for train and eval
 DataLoader:
   Train:
     dataset:
-        name: ImageNetDataset
-        image_root: "./dataset/ILSVRC2012/"
-        cls_label_path: "./dataset/ILSVRC2012/train_list.txt"
-        transform_ops:
-          - RandCropImage:
-              size: 224
-          - RandFlipImage:
-              flip_code: 1
-          - NormalizeImage:
-              scale: 0.00392157
-              mean: [0.485, 0.456, 0.406]
-              std: [0.229, 0.224, 0.225]
-              order: ''
+      name: "ImageNetDataset"
+      image_root: "./dataset/ILSVRC2012/"
+      cls_label_path: "./dataset/ILSVRC2012/train_list.txt"
+      transform_ops:
+        - RandCropImage:
+            size: 224
+        - RandFlipImage:
+            flip_code: 1
+        - NormalizeImage:
+            scale: 1.0/255.0
+            mean: [0.485, 0.456, 0.406]
+            std: [0.229, 0.224, 0.225]
+            order: ''
 
     sampler:
-        name: DistributedBatchSampler
-        batch_size: 512
-        drop_last: False
-        shuffle: True
+      name: "DistributedBatchSampler"
+      batch_size: 512
+      drop_last: False
+      shuffle: True
     loader:
-        num_workers: 6
-        use_shared_memory: True
+      num_workers: 4
+      use_shared_memory: True
 
   Eval:
     # TOTO: modify to the latest trainer
     dataset: 
-        name: ImageNetDataset
-        image_root: "./dataset/ILSVRC2012/"
-        cls_label_path: "./dataset/ILSVRC2012/val_list.txt"
-        transform_ops:
-          - ResizeImage:
-              resize_short: 256
-          - CropImage:
-              size: 224
-          - NormalizeImage:
-              scale: 0.00392157
-              mean: [0.485, 0.456, 0.406]
-              std: [0.229, 0.224, 0.225]
-              order: ''
+      name: "ImageNetDataset"
+      image_root: "./dataset/ILSVRC2012/"
+      cls_label_path: "./dataset/ILSVRC2012/val_list.txt"
+      transform_ops:
+        - ResizeImage:
+            resize_short: 256
+        - CropImage:
+            size: 224
+        - NormalizeImage:
+            scale: 1.0/255.0
+            mean: [0.485, 0.456, 0.406]
+            std: [0.229, 0.224, 0.225]
+            order: ''
     sampler:
-        name: DistributedBatchSampler
-        batch_size: 64
-        drop_last: False
-        shuffle: False
+      name: "DistributedBatchSampler"
+      batch_size: 64
+      drop_last: False
+      shuffle: False
     loader:
-        num_workers: 6
-        use_shared_memory: True
+      num_workers: 4
+      use_shared_memory: True
 
 Infer:
   infer_imgs: "docs/images/whl/demo.jpg"
   batch_size: 10
   transforms:
-      - DecodeImage:
-          to_rgb: True
-          channel_first: False
-      - ResizeImage:
-          resize_short: 256
-      - CropImage:
-          size: 224
-      - NormalizeImage:
-          scale: 1.0/255.0
-          mean: [0.485, 0.456, 0.406]
-          std: [0.229, 0.224, 0.225]
-          order: ''
-      - ToCHWImage:
+    - DecodeImage:
+        to_rgb: True
+        channel_first: False
+    - ResizeImage:
+        resize_short: 256
+    - CropImage:
+        size: 224
+    - NormalizeImage:
+        scale: 1.0/255.0
+        mean: [0.485, 0.456, 0.406]
+        std: [0.229, 0.224, 0.225]
+        order: ''
+    - ToCHWImage:
   PostProcess:
-    name: Topk
+    name: "Topk"
     topk: 5
     class_id_map_file: "ppcls/utils/imagenet1k_label_list.txt"
 
 Metric:
-    Train:
+  Train:
     - TopkAcc:
         topk: [1, 5]
-    Eval:
+  Eval:
     - TopkAcc:
         topk: [1, 5]

ppcls/configs/ImageNet/MobileNetV3/MobileNetV3_small_x0_35.yaml

@@ -31,94 +31,94 @@ Loss:
 
 
 Optimizer:
-  name: Momentum
+  name: "Momentum"
   momentum: 0.9
   lr:
-    name: Cosine
+    name: "Cosine"
     learning_rate: 1.3
   regularizer:
     name: 'L2'
-    coeff: 0.00002
+    coeff: 0.00001
 
 
 # data loader for train and eval
 DataLoader:
   Train:
     dataset:
-        name: ImageNetDataset
-        image_root: "./dataset/ILSVRC2012/"
-        cls_label_path: "./dataset/ILSVRC2012/train_list.txt"
-        transform_ops:
-          - RandCropImage:
-              size: 224
-          - RandFlipImage:
-              flip_code: 1
-          - NormalizeImage:
-              scale: 0.00392157
-              mean: [0.485, 0.456, 0.406]
-              std: [0.229, 0.224, 0.225]
-              order: ''
+      name: "ImageNetDataset"
+      image_root: "./dataset/ILSVRC2012/"
+      cls_label_path: "./dataset/ILSVRC2012/train_list.txt"
+      transform_ops:
+        - RandCropImage:
+            size: 224
+        - RandFlipImage:
+            flip_code: 1
+        - NormalizeImage:
+            scale: 1.0/255.0
+            mean: [0.485, 0.456, 0.406]
+            std: [0.229, 0.224, 0.225]
+            order: ''
 
     sampler:
-        name: DistributedBatchSampler
-        batch_size: 512
-        drop_last: False
-        shuffle: True
+      name: "DistributedBatchSampler"
+      batch_size: 512
+      drop_last: False
+      shuffle: True
     loader:
-        num_workers: 6
-        use_shared_memory: True
+      num_workers: 4
+      use_shared_memory: True
 
   Eval:
     # TOTO: modify to the latest trainer
     dataset: 
-        name: ImageNetDataset
-        image_root: "./dataset/ILSVRC2012/"
-        cls_label_path: "./dataset/ILSVRC2012/val_list.txt"
-        transform_ops:
-          - ResizeImage:
-              resize_short: 256
-          - CropImage:
-              size: 224
-          - NormalizeImage:
-              scale: 0.00392157
-              mean: [0.485, 0.456, 0.406]
-              std: [0.229, 0.224, 0.225]
-              order: ''
+      name: "ImageNetDataset"
+      image_root: "./dataset/ILSVRC2012/"
+      cls_label_path: "./dataset/ILSVRC2012/val_list.txt"
+      transform_ops:
+        - ResizeImage:
+            resize_short: 256
+        - CropImage:
+            size: 224
+        - NormalizeImage:
+            scale: 1.0/255.0
+            mean: [0.485, 0.456, 0.406]
+            std: [0.229, 0.224, 0.225]
+            order: ''
     sampler:
-        name: DistributedBatchSampler
-        batch_size: 64
-        drop_last: False
-        shuffle: False
+      name: "DistributedBatchSampler"
+      batch_size: 64
+      drop_last: False
+      shuffle: False
     loader:
-        num_workers: 6
-        use_shared_memory: True
+      num_workers: 4
+      use_shared_memory: True
 
 Infer:
   infer_imgs: "docs/images/whl/demo.jpg"
   batch_size: 10
   transforms:
-      - DecodeImage:
-          to_rgb: True
-          channel_first: False
-      - ResizeImage:
-          resize_short: 256
-      - CropImage:
-          size: 224
-      - NormalizeImage:
-          scale: 1.0/255.0
-          mean: [0.485, 0.456, 0.406]
-          std: [0.229, 0.224, 0.225]
-          order: ''
-      - ToCHWImage:
+    - DecodeImage:
+        to_rgb: True
+        channel_first: False
+    - ResizeImage:
+        resize_short: 256
+    - CropImage:
+        size: 224
+    - NormalizeImage:
+        scale: 1.0/255.0
+        mean: [0.485, 0.456, 0.406]
+        std: [0.229, 0.224, 0.225]
+        order: ''
+    - ToCHWImage:
   PostProcess:
-    name: Topk
+    name: "Topk"
     topk: 5
     class_id_map_file: "ppcls/utils/imagenet1k_label_list.txt"
 
 Metric:
-    Train:
+  Train:
     - TopkAcc:
         topk: [1, 5]
-    Eval:
+  Eval:
     - TopkAcc:
         topk: [1, 5]

ppcls/configs/ImageNet/MobileNetV3/MobileNetV3_small_x0_5.yaml

@@ -31,94 +31,94 @@ Loss:
 
 
 Optimizer:
-  name: Momentum
+  name: "Momentum"
   momentum: 0.9
   lr:
-    name: Cosine
+    name: "Cosine"
     learning_rate: 1.3
   regularizer:
     name: 'L2'
-    coeff: 0.00002
+    coeff: 0.00001
 
 
 # data loader for train and eval
 DataLoader:
   Train:
     dataset:
-        name: ImageNetDataset
-        image_root: "./dataset/ILSVRC2012/"
-        cls_label_path: "./dataset/ILSVRC2012/train_list.txt"
-        transform_ops:
-          - RandCropImage:
-              size: 224
-          - RandFlipImage:
-              flip_code: 1
-          - NormalizeImage:
-              scale: 0.00392157
-              mean: [0.485, 0.456, 0.406]
-              std: [0.229, 0.224, 0.225]
-              order: ''
+      name: "ImageNetDataset"
+      image_root: "./dataset/ILSVRC2012/"
+      cls_label_path: "./dataset/ILSVRC2012/train_list.txt"
+      transform_ops:
+        - RandCropImage:
+            size: 224
+        - RandFlipImage:
+            flip_code: 1
+        - NormalizeImage:
+            scale: 1.0/255.0
+            mean: [0.485, 0.456, 0.406]
+            std: [0.229, 0.224, 0.225]
+            order: ''
 
     sampler:
-        name: DistributedBatchSampler
-        batch_size: 512
-        drop_last: False
-        shuffle: True
+      name: "DistributedBatchSampler"
+      batch_size: 512
+      drop_last: False
+      shuffle: True
     loader:
-        num_workers: 6
-        use_shared_memory: True
+      num_workers: 4
+      use_shared_memory: True
 
   Eval:
     # TOTO: modify to the latest trainer
     dataset: 
-        name: ImageNetDataset
-        image_root: "./dataset/ILSVRC2012/"
-        cls_label_path: "./dataset/ILSVRC2012/val_list.txt"
-        transform_ops:
-          - ResizeImage:
-              resize_short: 256
-          - CropImage:
-              size: 224
-          - NormalizeImage:
-              scale: 0.00392157
-              mean: [0.485, 0.456, 0.406]
-              std: [0.229, 0.224, 0.225]
-              order: ''
+      name: "ImageNetDataset"
+      image_root: "./dataset/ILSVRC2012/"
+      cls_label_path: "./dataset/ILSVRC2012/val_list.txt"
+      transform_ops:
+        - ResizeImage:
+            resize_short: 256
+        - CropImage:
+            size: 224
+        - NormalizeImage:
+            scale: 1.0/255.0
+            mean: [0.485, 0.456, 0.406]
+            std: [0.229, 0.224, 0.225]
+            order: ''
     sampler:
-        name: DistributedBatchSampler
-        batch_size: 64
-        drop_last: False
-        shuffle: False
+      name: "DistributedBatchSampler"
+      batch_size: 64
+      drop_last: False
+      shuffle: False
     loader:
-        num_workers: 6
-        use_shared_memory: True
+      num_workers: 4
+      use_shared_memory: True
 
 Infer:
   infer_imgs: "docs/images/whl/demo.jpg"
   batch_size: 10
   transforms:
-      - DecodeImage:
-          to_rgb: True
-          channel_first: False
-      - ResizeImage:
-          resize_short: 256
-      - CropImage:
-          size: 224
-      - NormalizeImage:
-          scale: 1.0/255.0
-          mean: [0.485, 0.456, 0.406]
-          std: [0.229, 0.224, 0.225]
-          order: ''
-      - ToCHWImage:
+    - DecodeImage:
+        to_rgb: True
+        channel_first: False
+    - ResizeImage:
+        resize_short: 256
+    - CropImage:
+        size: 224
+    - NormalizeImage:
+        scale: 1.0/255.0
+        mean: [0.485, 0.456, 0.406]
+        std: [0.229, 0.224, 0.225]
+        order: ''
+    - ToCHWImage:
   PostProcess:
-    name: Topk
+    name: "Topk"
     topk: 5
     class_id_map_file: "ppcls/utils/imagenet1k_label_list.txt"
 
 Metric:
-    Train:
+  Train:
     - TopkAcc:
         topk: [1, 5]
-    Eval:
+  Eval:
     - TopkAcc:
         topk: [1, 5]

ppcls/configs/ImageNet/MobileNetV3/MobileNetV3_small_x0_75.yaml

@@ -31,10 +31,10 @@ Loss:
 
 
 Optimizer:
-  name: Momentum
+  name: "Momentum"
   momentum: 0.9
   lr:
-    name: Cosine
+    name: "Cosine"
     learning_rate: 1.3
   regularizer:
     name: 'L2'
@@ -45,80 +45,80 @@ Optimizer:
 DataLoader:
   Train:
     dataset:
-        name: ImageNetDataset
-        image_root: "./dataset/ILSVRC2012/"
-        cls_label_path: "./dataset/ILSVRC2012/train_list.txt"
-        transform_ops:
-          - RandCropImage:
-              size: 224
-          - RandFlipImage:
-              flip_code: 1
-          - NormalizeImage:
-              scale: 0.00392157
-              mean: [0.485, 0.456, 0.406]
-              std: [0.229, 0.224, 0.225]
-              order: ''
+      name: "ImageNetDataset"
+      image_root: "./dataset/ILSVRC2012/"
+      cls_label_path: "./dataset/ILSVRC2012/train_list.txt"
+      transform_ops:
+        - RandCropImage:
+            size: 224
+        - RandFlipImage:
+            flip_code: 1
+        - NormalizeImage:
+            scale: 1.0/255.0
+            mean: [0.485, 0.456, 0.406]
+            std: [0.229, 0.224, 0.225]
+            order: ''
 
     sampler:
-        name: DistributedBatchSampler
-        batch_size: 512
-        drop_last: False
-        shuffle: True
+      name: "DistributedBatchSampler"
+      batch_size: 512
+      drop_last: False
+      shuffle: True
     loader:
-        num_workers: 6
-        use_shared_memory: True
+      num_workers: 4
+      use_shared_memory: True
 
   Eval:
     # TOTO: modify to the latest trainer
     dataset: 
-        name: ImageNetDataset
-        image_root: "./dataset/ILSVRC2012/"
-        cls_label_path: "./dataset/ILSVRC2012/val_list.txt"
-        transform_ops:
-          - ResizeImage:
-              resize_short: 256
-          - CropImage:
-              size: 224
-          - NormalizeImage:
-              scale: 0.00392157
-              mean: [0.485, 0.456, 0.406]
-              std: [0.229, 0.224, 0.225]
-              order: ''
+      name: "ImageNetDataset"
+      image_root: "./dataset/ILSVRC2012/"
+      cls_label_path: "./dataset/ILSVRC2012/val_list.txt"
+      transform_ops:
+        - ResizeImage:
+            resize_short: 256
+        - CropImage:
+            size: 224
+        - NormalizeImage:
+            scale: 1.0/255.0
+            mean: [0.485, 0.456, 0.406]
+            std: [0.229, 0.224, 0.225]
+            order: ''
     sampler:
-        name: DistributedBatchSampler
-        batch_size: 64
-        drop_last: False
-        shuffle: False
+      name: "DistributedBatchSampler"
+      batch_size: 64
+      drop_last: False
+      shuffle: False
     loader:
-        num_workers: 6
-        use_shared_memory: True
+      num_workers: 4
+      use_shared_memory: True
 
 Infer:
   infer_imgs: "docs/images/whl/demo.jpg"
   batch_size: 10
   transforms:
-      - DecodeImage:
-          to_rgb: True
-          channel_first: False
-      - ResizeImage:
-          resize_short: 256
-      - CropImage:
-          size: 224
-      - NormalizeImage:
-          scale: 1.0/255.0
-          mean: [0.485, 0.456, 0.406]
-          std: [0.229, 0.224, 0.225]
-          order: ''
-      - ToCHWImage:
+    - DecodeImage:
+        to_rgb: True
+        channel_first: False
+    - ResizeImage:
+        resize_short: 256
+    - CropImage:
+        size: 224
+    - NormalizeImage:
+        scale: 1.0/255.0
+        mean: [0.485, 0.456, 0.406]
+        std: [0.229, 0.224, 0.225]
+        order: ''
+    - ToCHWImage:
   PostProcess:
-    name: Topk
+    name: "Topk"
     topk: 5
     class_id_map_file: "ppcls/utils/imagenet1k_label_list.txt"
 
 Metric:
-    Train:
+  Train:
     - TopkAcc:
         topk: [1, 5]
-    Eval:
+  Eval:
     - TopkAcc:
         topk: [1, 5]

ppcls/configs/ImageNet/MobileNetV3/MobileNetV3_small_x1_0.yaml

@@ -31,10 +31,10 @@ Loss:
 
 
 Optimizer:
-  name: Momentum
+  name: "Momentum"
   momentum: 0.9
   lr:
-    name: Cosine
+    name: "Cosine"
     learning_rate: 1.3
   regularizer:
     name: 'L2'
@@ -45,80 +45,80 @@ Optimizer:
 DataLoader:
   Train:
     dataset:
-        name: ImageNetDataset
-        image_root: "./dataset/ILSVRC2012/"
-        cls_label_path: "./dataset/ILSVRC2012/train_list.txt"
-        transform_ops:
-          - RandCropImage:
-              size: 224
-          - RandFlipImage:
-              flip_code: 1
-          - NormalizeImage:
-              scale: 0.00392157
-              mean: [0.485, 0.456, 0.406]
-              std: [0.229, 0.224, 0.225]
-              order: ''
+      name: "ImageNetDataset"
+      image_root: "./dataset/ILSVRC2012/"
+      cls_label_path: "./dataset/ILSVRC2012/train_list.txt"
+      transform_ops:
+        - RandCropImage:
+            size: 224
+        - RandFlipImage:
+            flip_code: 1
+        - NormalizeImage:
+            scale: 1.0/255.0
+            mean: [0.485, 0.456, 0.406]
+            std: [0.229, 0.224, 0.225]
+            order: ''
 
     sampler:
-        name: DistributedBatchSampler
-        batch_size: 512
-        drop_last: False
-        shuffle: True
+      name: "DistributedBatchSampler"
+      batch_size: 512
+      drop_last: False
+      shuffle: True
     loader:
-        num_workers: 6
-        use_shared_memory: True
+      num_workers: 4
+      use_shared_memory: True
 
   Eval:
     # TOTO: modify to the latest trainer
     dataset: 
-        name: ImageNetDataset
-        image_root: "./dataset/ILSVRC2012/"
-        cls_label_path: "./dataset/ILSVRC2012/val_list.txt"
-        transform_ops:
-          - ResizeImage:
-              resize_short: 256
-          - CropImage:
-              size: 224
-          - NormalizeImage:
-              scale: 0.00392157
-              mean: [0.485, 0.456, 0.406]
-              std: [0.229, 0.224, 0.225]
-              order: ''
+      name: "ImageNetDataset"
+      image_root: "./dataset/ILSVRC2012/"
+      cls_label_path: "./dataset/ILSVRC2012/val_list.txt"
+      transform_ops:
+        - ResizeImage:
+            resize_short: 256
+        - CropImage:
+            size: 224
+        - NormalizeImage:
+            scale: 1.0/255.0
+            mean: [0.485, 0.456, 0.406]
+            std: [0.229, 0.224, 0.225]
+            order: ''
     sampler:
-        name: DistributedBatchSampler
-        batch_size: 64
-        drop_last: False
-        shuffle: False
+      name: "DistributedBatchSampler"
+      batch_size: 64
+      drop_last: False
+      shuffle: False
     loader:
-        num_workers: 6
-        use_shared_memory: True
+      num_workers: 4
+      use_shared_memory: True
 
 Infer:
   infer_imgs: "docs/images/whl/demo.jpg"
   batch_size: 10
   transforms:
-      - DecodeImage:
-          to_rgb: True
-          channel_first: False
-      - ResizeImage:
-          resize_short: 256
-      - CropImage:
-          size: 224
-      - NormalizeImage:
-          scale: 1.0/255.0
-          mean: [0.485, 0.456, 0.406]
-          std: [0.229, 0.224, 0.225]
-          order: ''
-      - ToCHWImage:
+    - DecodeImage:
+        to_rgb: True
+        channel_first: False
+    - ResizeImage:
+        resize_short: 256
+    - CropImage:
+        size: 224
+    - NormalizeImage:
+        scale: 1.0/255.0
+        mean: [0.485, 0.456, 0.406]
+        std: [0.229, 0.224, 0.225]
+        order: ''
+    - ToCHWImage:
   PostProcess:
-    name: Topk
+    name: "Topk"
     topk: 5
     class_id_map_file: "ppcls/utils/imagenet1k_label_list.txt"
 
 Metric:
-    Train:
+  Train:
     - TopkAcc:
         topk: [1, 5]
-    Eval:
+  Eval:
     - TopkAcc:
         topk: [1, 5]

ppcls/configs/ImageNet/MobileNetV3/MobileNetV3_small_x1_25.yaml

@@ -31,10 +31,10 @@ Loss:
 
 
 Optimizer:
-  name: Momentum
+  name: "Momentum"
   momentum: 0.9
   lr:
-    name: Cosine
+    name: "Cosine"
     learning_rate: 1.3
   regularizer:
     name: 'L2'
@@ -45,80 +45,80 @@ Optimizer:
 DataLoader:
   Train:
     dataset:
-        name: ImageNetDataset
-        image_root: "./dataset/ILSVRC2012/"
-        cls_label_path: "./dataset/ILSVRC2012/train_list.txt"
-        transform_ops:
-          - RandCropImage:
-              size: 224
-          - RandFlipImage:
-              flip_code: 1
-          - NormalizeImage:
-              scale: 0.00392157
-              mean: [0.485, 0.456, 0.406]
-              std: [0.229, 0.224, 0.225]
-              order: ''
+      name: "ImageNetDataset"
+      image_root: "./dataset/ILSVRC2012/"
+      cls_label_path: "./dataset/ILSVRC2012/train_list.txt"
+      transform_ops:
+        - RandCropImage:
+            size: 224
+        - RandFlipImage:
+            flip_code: 1
+        - NormalizeImage:
+            scale: 1.0/255.0
+            mean: [0.485, 0.456, 0.406]
+            std: [0.229, 0.224, 0.225]
+            order: ''
 
     sampler:
-        name: DistributedBatchSampler
-        batch_size: 512
-        drop_last: False
-        shuffle: True
+      name: "DistributedBatchSampler"
+      batch_size: 512
+      drop_last: False
+      shuffle: True
     loader:
-        num_workers: 6
-        use_shared_memory: True
+      num_workers: 4
+      use_shared_memory: True
 
   Eval:
     # TOTO: modify to the latest trainer
     dataset: 
-        name: ImageNetDataset
-        image_root: "./dataset/ILSVRC2012/"
-        cls_label_path: "./dataset/ILSVRC2012/val_list.txt"
-        transform_ops:
-          - ResizeImage:
-              resize_short: 256
-          - CropImage:
-              size: 224
-          - NormalizeImage:
-              scale: 0.00392157
-              mean: [0.485, 0.456, 0.406]
-              std: [0.229, 0.224, 0.225]
-              order: ''
+      name: "ImageNetDataset"
+      image_root: "./dataset/ILSVRC2012/"
+      cls_label_path: "./dataset/ILSVRC2012/val_list.txt"
+      transform_ops:
+        - ResizeImage:
+            resize_short: 256
+        - CropImage:
+            size: 224
+        - NormalizeImage:
+            scale: 1.0/255.0
+            mean: [0.485, 0.456, 0.406]
+            std: [0.229, 0.224, 0.225]
+            order: ''
     sampler:
-        name: DistributedBatchSampler
-        batch_size: 64
-        drop_last: False
-        shuffle: False
+      name: "DistributedBatchSampler"
+      batch_size: 64
+      drop_last: False
+      shuffle: False
     loader:
-        num_workers: 6
-        use_shared_memory: True
+      num_workers: 4
+      use_shared_memory: True
 
 Infer:
   infer_imgs: "docs/images/whl/demo.jpg"
   batch_size: 10
   transforms:
-      - DecodeImage:
-          to_rgb: True
-          channel_first: False
-      - ResizeImage:
-          resize_short: 256
-      - CropImage:
-          size: 224
-      - NormalizeImage:
-          scale: 1.0/255.0
-          mean: [0.485, 0.456, 0.406]
-          std: [0.229, 0.224, 0.225]
-          order: ''
-      - ToCHWImage:
+    - DecodeImage:
+        to_rgb: True
+        channel_first: False
+    - ResizeImage:
+        resize_short: 256
+    - CropImage:
+        size: 224
+    - NormalizeImage:
+        scale: 1.0/255.0
+        mean: [0.485, 0.456, 0.406]
+        std: [0.229, 0.224, 0.225]
+        order: ''
+    - ToCHWImage:
   PostProcess:
-    name: Topk
+    name: "Topk"
     topk: 5
     class_id_map_file: "ppcls/utils/imagenet1k_label_list.txt"
 
 Metric:
-    Train:
+  Train:
     - TopkAcc:
         topk: [1, 5]
-    Eval:
+  Eval:
     - TopkAcc:
         topk: [1, 5]

ppcls/configs/ImageNet/ResNet/ResNet101.yaml

@@ -66,7 +66,7 @@ DataLoader:
       drop_last: False
       shuffle: True
     loader:
-      num_workers: 6
+      num_workers: 4
       use_shared_memory: True
 
   Eval:
@@ -91,7 +91,7 @@ DataLoader:
       drop_last: False
       shuffle: False
     loader:
-      num_workers: 6
+      num_workers: 4
       use_shared_memory: True
 
 Infer:

ppcls/configs/ImageNet/ResNet/ResNet101_vd.yaml

@@ -68,7 +68,7 @@ DataLoader:
       drop_last: False
       shuffle: True
     loader:
-      num_workers: 6
+      num_workers: 4
       use_shared_memory: True
 
   Eval:
@@ -93,7 +93,7 @@ DataLoader:
       drop_last: False
       shuffle: False
     loader:
-      num_workers: 6
+      num_workers: 4
       use_shared_memory: True
 
 Infer:

ppcls/configs/ImageNet/ResNet/ResNet152.yaml

@@ -66,7 +66,7 @@ DataLoader:
       drop_last: False
       shuffle: True
     loader:
-      num_workers: 6
+      num_workers: 4
       use_shared_memory: True
 
   Eval:
@@ -91,7 +91,7 @@ DataLoader:
       drop_last: False
       shuffle: False
     loader:
-      num_workers: 6
+      num_workers: 4
       use_shared_memory: True
 
 Infer:

ppcls/configs/ImageNet/ResNet/ResNet152_vd.yaml

@@ -68,7 +68,7 @@ DataLoader:
       drop_last: False
       shuffle: True
     loader:
-      num_workers: 6
+      num_workers: 4
       use_shared_memory: True
 
   Eval:
@@ -93,7 +93,7 @@ DataLoader:
       drop_last: False
       shuffle: False
     loader:
-      num_workers: 6
+      num_workers: 4
       use_shared_memory: True
 
 Infer:

ppcls/configs/ImageNet/ResNet/ResNet18.yaml

@@ -66,7 +66,7 @@ DataLoader:
       drop_last: False
       shuffle: True
     loader:
-      num_workers: 6
+      num_workers: 4
       use_shared_memory: True
 
   Eval:
@@ -91,7 +91,7 @@ DataLoader:
       drop_last: False
       shuffle: False
     loader:
-      num_workers: 6
+      num_workers: 4
       use_shared_memory: True
 
 Infer:

ppcls/configs/ImageNet/ResNet/ResNet18_vd.yaml

@@ -68,7 +68,7 @@ DataLoader:
       drop_last: False
       shuffle: True
     loader:
-      num_workers: 6
+      num_workers: 4
       use_shared_memory: True
 
   Eval:
@@ -93,7 +93,7 @@ DataLoader:
       drop_last: False
       shuffle: False
     loader:
-      num_workers: 6
+      num_workers: 4
       use_shared_memory: True
 
 Infer:

ppcls/configs/ImageNet/ResNet/ResNet200_vd.yaml

@@ -68,7 +68,7 @@ DataLoader:
       drop_last: False
       shuffle: True
     loader:
-      num_workers: 6
+      num_workers: 4
       use_shared_memory: True
 
   Eval:
@@ -93,7 +93,7 @@ DataLoader:
       drop_last: False
       shuffle: False
     loader:
-      num_workers: 6
+      num_workers: 4
       use_shared_memory: True
 
 Infer:

ppcls/configs/ImageNet/ResNet/ResNet34.yaml

@@ -66,7 +66,7 @@ DataLoader:
       drop_last: False
       shuffle: True
     loader:
-      num_workers: 6
+      num_workers: 4
       use_shared_memory: True
 
   Eval:
@@ -91,7 +91,7 @@ DataLoader:
       drop_last: False
       shuffle: False
     loader:
-      num_workers: 6
+      num_workers: 4
       use_shared_memory: True
 
 Infer:

ppcls/configs/ImageNet/ResNet/ResNet34_vd.yaml

@@ -68,7 +68,7 @@ DataLoader:
       drop_last: False
       shuffle: True
     loader:
-      num_workers: 6
+      num_workers: 4
       use_shared_memory: True
 
   Eval:
@@ -93,7 +93,7 @@ DataLoader:
       drop_last: False
       shuffle: False
     loader:
-      num_workers: 6
+      num_workers: 4
       use_shared_memory: True
 
 Infer:

ppcls/configs/ImageNet/ResNet/ResNet50.yaml

@@ -66,7 +66,7 @@ DataLoader:
       drop_last: False
       shuffle: True
     loader:
-      num_workers: 6
+      num_workers: 4
       use_shared_memory: True
 
   Eval:
@@ -91,7 +91,7 @@ DataLoader:
       drop_last: False
       shuffle: False
     loader:
-      num_workers: 6
+      num_workers: 4
       use_shared_memory: True
 
 Infer:

ppcls/configs/ImageNet/ResNet/ResNet50_vd.yaml

@@ -68,7 +68,7 @@ DataLoader:
       drop_last: False
       shuffle: True
     loader:
-      num_workers: 6
+      num_workers: 4
       use_shared_memory: True
 
   Eval:
@@ -93,7 +93,7 @@ DataLoader:
       drop_last: False
       shuffle: False
     loader:
-      num_workers: 6
+      num_workers: 4
       use_shared_memory: True
 
 Infer:

ppcls/configs/ImageNet/VGG/VGG11.yaml

@@ -30,10 +30,10 @@ Loss:
 
 
 Optimizer:
-  name: Momentum
+  name: "Momentum"
   momentum: 0.9
   lr:
-    name: Cosine
+    name: "Cosine"
     learning_rate: 0.1
   regularizer:
     name: 'L2'
@@ -44,80 +44,80 @@ Optimizer:
 DataLoader:
   Train:
     dataset:
-        name: ImageNetDataset
-        image_root: "./dataset/ILSVRC2012/"
-        cls_label_path: "./dataset/ILSVRC2012/train_list.txt"
-        transform_ops:
-          - RandCropImage:
-              size: 224
-          - RandFlipImage:
-              flip_code: 1
-          - NormalizeImage:
-              scale: 0.00392157
-              mean: [0.485, 0.456, 0.406]
-              std: [0.229, 0.224, 0.225]
-              order: ''
+      name: "ImageNetDataset"
+      image_root: "./dataset/ILSVRC2012/"
+      cls_label_path: "./dataset/ILSVRC2012/train_list.txt"
+      transform_ops:
+        - RandCropImage:
+            size: 224
+        - RandFlipImage:
+            flip_code: 1
+        - NormalizeImage:
+            scale: 1.0/255.0
+            mean: [0.485, 0.456, 0.406]
+            std: [0.229, 0.224, 0.225]
+            order: ''
 
     sampler:
-        name: DistributedBatchSampler
-        batch_size: 64
-        drop_last: False
-        shuffle: True
+      name: "DistributedBatchSampler"
+      batch_size: 128
+      drop_last: False
+      shuffle: True
     loader:
-        num_workers: 6
-        use_shared_memory: True
+      num_workers: 4
+      use_shared_memory: True
 
   Eval:
     # TOTO: modify to the latest trainer
     dataset: 
-        name: ImageNetDataset
-        image_root: "./dataset/ILSVRC2012/"
-        cls_label_path: "./dataset/ILSVRC2012/val_list.txt"
-        transform_ops:
-          - ResizeImage:
-              resize_short: 256
-          - CropImage:
-              size: 224
-          - NormalizeImage:
-              scale: 0.00392157
-              mean: [0.485, 0.456, 0.406]
-              std: [0.229, 0.224, 0.225]
-              order: ''
+      name: "ImageNetDataset"
+      image_root: "./dataset/ILSVRC2012/"
+      cls_label_path: "./dataset/ILSVRC2012/val_list.txt"
+      transform_ops:
+        - ResizeImage:
+            resize_short: 256
+        - CropImage:
+            size: 224
+        - NormalizeImage:
+            scale: 1.0/255.0
+            mean: [0.485, 0.456, 0.406]
+            std: [0.229, 0.224, 0.225]
+            order: ''
     sampler:
-        name: DistributedBatchSampler
-        batch_size: 64
-        drop_last: False
-        shuffle: False
+      name: "DistributedBatchSampler"
+      batch_size: 64
+      drop_last: False
+      shuffle: False
     loader:
-        num_workers: 6
-        use_shared_memory: True
+      num_workers: 4
+      use_shared_memory: True
 
 Infer:
   infer_imgs: "docs/images/whl/demo.jpg"
   batch_size: 10
   transforms:
-      - DecodeImage:
-          to_rgb: True
-          channel_first: False
-      - ResizeImage:
-          resize_short: 256
-      - CropImage:
-          size: 224
-      - NormalizeImage:
-          scale: 1.0/255.0
-          mean: [0.485, 0.456, 0.406]
-          std: [0.229, 0.224, 0.225]
-          order: ''
-      - ToCHWImage:
+    - DecodeImage:
+        to_rgb: True
+        channel_first: False
+    - ResizeImage:
+        resize_short: 256
+    - CropImage:
+        size: 224
+    - NormalizeImage:
+        scale: 1.0/255.0
+        mean: [0.485, 0.456, 0.406]
+        std: [0.229, 0.224, 0.225]
+        order: ''
+    - ToCHWImage:
   PostProcess:
-    name: Topk
+    name: "Topk"
     topk: 5
     class_id_map_file: "ppcls/utils/imagenet1k_label_list.txt"
 
 Metric:
-    Train:
+  Train:
     - TopkAcc:
         topk: [1, 5]
-    Eval:
+  Eval:
     - TopkAcc:
         topk: [1, 5]

ppcls/configs/ImageNet/VGG/VGG13.yaml

@@ -30,10 +30,10 @@ Loss:
 
 
 Optimizer:
-  name: Momentum
+  name: "Momentum"
   momentum: 0.9
   lr:
-    name: Cosine
+    name: "Cosine"
     learning_rate: 0.01
   regularizer:
     name: 'L2'
@@ -44,80 +44,80 @@ Optimizer:
 DataLoader:
   Train:
     dataset:
-        name: ImageNetDataset
-        image_root: "./dataset/ILSVRC2012/"
-        cls_label_path: "./dataset/ILSVRC2012/train_list.txt"
-        transform_ops:
-          - RandCropImage:
-              size: 224
-          - RandFlipImage:
-              flip_code: 1
-          - NormalizeImage:
-              scale: 0.00392157
-              mean: [0.485, 0.456, 0.406]
-              std: [0.229, 0.224, 0.225]
-              order: ''
+      name: "ImageNetDataset"
+      image_root: "./dataset/ILSVRC2012/"
+      cls_label_path: "./dataset/ILSVRC2012/train_list.txt"
+      transform_ops:
+        - RandCropImage:
+            size: 224
+        - RandFlipImage:
+            flip_code: 1
+        - NormalizeImage:
+            scale: 1.0/255.0
+            mean: [0.485, 0.456, 0.406]
+            std: [0.229, 0.224, 0.225]
+            order: ''
 
     sampler:
-        name: DistributedBatchSampler
-        batch_size: 64
-        drop_last: False
-        shuffle: True
+      name: "DistributedBatchSampler"
+      batch_size: 64
+      drop_last: False
+      shuffle: True
     loader:
-        num_workers: 6
-        use_shared_memory: True
+      num_workers: 4
+      use_shared_memory: True
 
   Eval:
     # TOTO: modify to the latest trainer
     dataset: 
-        name: ImageNetDataset
-        image_root: "./dataset/ILSVRC2012/"
-        cls_label_path: "./dataset/ILSVRC2012/val_list.txt"
-        transform_ops:
-          - ResizeImage:
-              resize_short: 256
-          - CropImage:
-              size: 224
-          - NormalizeImage:
-              scale: 0.00392157
-              mean: [0.485, 0.456, 0.406]
-              std: [0.229, 0.224, 0.225]
-              order: ''
+      name: "ImageNetDataset"
+      image_root: "./dataset/ILSVRC2012/"
+      cls_label_path: "./dataset/ILSVRC2012/val_list.txt"
+      transform_ops:
+        - ResizeImage:
+            resize_short: 256
+        - CropImage:
+            size: 224
+        - NormalizeImage:
+            scale: 1.0/255.0
+            mean: [0.485, 0.456, 0.406]
+            std: [0.229, 0.224, 0.225]
+            order: ''
     sampler:
-        name: DistributedBatchSampler
-        batch_size: 64
-        drop_last: False
-        shuffle: False
+      name: "DistributedBatchSampler"
+      batch_size: 64
+      drop_last: False
+      shuffle: False
     loader:
-        num_workers: 6
-        use_shared_memory: True
+      num_workers: 4
+      use_shared_memory: True
 
 Infer:
   infer_imgs: "docs/images/whl/demo.jpg"
   batch_size: 10
   transforms:
-      - DecodeImage:
-          to_rgb: True
-          channel_first: False
-      - ResizeImage:
-          resize_short: 256
-      - CropImage:
-          size: 224
-      - NormalizeImage:
-          scale: 1.0/255.0
-          mean: [0.485, 0.456, 0.406]
-          std: [0.229, 0.224, 0.225]
-          order: ''
-      - ToCHWImage:
+    - DecodeImage:
+        to_rgb: True
+        channel_first: False
+    - ResizeImage:
+        resize_short: 256
+    - CropImage:
+        size: 224
+    - NormalizeImage:
+        scale: 1.0/255.0
+        mean: [0.485, 0.456, 0.406]
+        std: [0.229, 0.224, 0.225]
+        order: ''
+    - ToCHWImage:
   PostProcess:
-    name: Topk
+    name: "Topk"
     topk: 5
     class_id_map_file: "ppcls/utils/imagenet1k_label_list.txt"
 
 Metric:
-    Train:
+  Train:
     - TopkAcc:
         topk: [1, 5]
-    Eval:
+  Eval:
     - TopkAcc:
         topk: [1, 5]

ppcls/configs/ImageNet/VGG/VGG16.yaml

@@ -30,10 +30,10 @@ Loss:
 
 
 Optimizer:
-  name: Momentum
+  name: "Momentum"
   momentum: 0.9
   lr:
-    name: Cosine
+    name: "Cosine"
     learning_rate: 0.01
   regularizer:
     name: 'L2'
@@ -44,80 +44,80 @@ Optimizer:
 DataLoader:
   Train:
     dataset:
-        name: ImageNetDataset
-        image_root: "./dataset/ILSVRC2012/"
-        cls_label_path: "./dataset/ILSVRC2012/train_list.txt"
-        transform_ops:
-          - RandCropImage:
-              size: 224
-          - RandFlipImage:
-              flip_code: 1
-          - NormalizeImage:
-              scale: 0.00392157
-              mean: [0.485, 0.456, 0.406]
-              std: [0.229, 0.224, 0.225]
-              order: ''
+      name: "ImageNetDataset"
+      image_root: "./dataset/ILSVRC2012/"
+      cls_label_path: "./dataset/ILSVRC2012/train_list.txt"
+      transform_ops:
+        - RandCropImage:
+            size: 224
+        - RandFlipImage:
+            flip_code: 1
+        - NormalizeImage:
+            scale: 1.0/255.0
+            mean: [0.485, 0.456, 0.406]
+            std: [0.229, 0.224, 0.225]
+            order: ''
 
     sampler:
-        name: DistributedBatchSampler
-        batch_size: 64
-        drop_last: False
-        shuffle: True
+      name: "DistributedBatchSampler"
+      batch_size: 64
+      drop_last: False
+      shuffle: True
     loader:
-        num_workers: 6
-        use_shared_memory: True
+      num_workers: 4
+      use_shared_memory: True
 
   Eval:
     # TOTO: modify to the latest trainer
     dataset: 
-        name: ImageNetDataset
-        image_root: "./dataset/ILSVRC2012/"
-        cls_label_path: "./dataset/ILSVRC2012/val_list.txt"
-        transform_ops:
-          - ResizeImage:
-              resize_short: 256
-          - CropImage:
-              size: 224
-          - NormalizeImage:
-              scale: 0.00392157
-              mean: [0.485, 0.456, 0.406]
-              std: [0.229, 0.224, 0.225]
-              order: ''
+      name: "ImageNetDataset"
+      image_root: "./dataset/ILSVRC2012/"
+      cls_label_path: "./dataset/ILSVRC2012/val_list.txt"
+      transform_ops:
+        - ResizeImage:
+            resize_short: 256
+        - CropImage:
+            size: 224
+        - NormalizeImage:
+            scale: 1.0/255.0
+            mean: [0.485, 0.456, 0.406]
+            std: [0.229, 0.224, 0.225]
+            order: ''
     sampler:
-        name: DistributedBatchSampler
-        batch_size: 64
-        drop_last: False
-        shuffle: False
+      name: "DistributedBatchSampler"
+      batch_size: 64
+      drop_last: False
+      shuffle: False
     loader:
-        num_workers: 6
-        use_shared_memory: True
+      num_workers: 4
+      use_shared_memory: True
 
 Infer:
   infer_imgs: "docs/images/whl/demo.jpg"
   batch_size: 10
   transforms:
-      - DecodeImage:
-          to_rgb: True
-          channel_first: False
-      - ResizeImage:
-          resize_short: 256
-      - CropImage:
-          size: 224
-      - NormalizeImage:
-          scale: 1.0/255.0
-          mean: [0.485, 0.456, 0.406]
-          std: [0.229, 0.224, 0.225]
-          order: ''
-      - ToCHWImage:
+    - DecodeImage:
+        to_rgb: True
+        channel_first: False
+    - ResizeImage:
+        resize_short: 256
+    - CropImage:
+        size: 224
+    - NormalizeImage:
+        scale: 1.0/255.0
+        mean: [0.485, 0.456, 0.406]
+        std: [0.229, 0.224, 0.225]
+        order: ''
+    - ToCHWImage:
   PostProcess:
-    name: Topk
+    name: "Topk"
     topk: 5
     class_id_map_file: "ppcls/utils/imagenet1k_label_list.txt"
 
 Metric:
-    Train:
+  Train:
     - TopkAcc:
         topk: [1, 5]
-    Eval:
+  Eval:
     - TopkAcc:
         topk: [1, 5]

ppcls/configs/ImageNet/VGG/VGG19.yaml

@@ -30,10 +30,10 @@ Loss:
 
 
 Optimizer:
-  name: Momentum
+  name: "Momentum"
   momentum: 0.9
   lr:
-    name: Cosine
+    name: "Cosine"
     learning_rate: 0.01
   regularizer:
     name: 'L2'
@@ -44,80 +44,80 @@ Optimizer:
 DataLoader:
   Train:
     dataset:
-        name: ImageNetDataset
-        image_root: "./dataset/ILSVRC2012/"
-        cls_label_path: "./dataset/ILSVRC2012/train_list.txt"
-        transform_ops:
-          - RandCropImage:
-              size: 224
-          - RandFlipImage:
-              flip_code: 1
-          - NormalizeImage:
-              scale: 0.00392157
-              mean: [0.485, 0.456, 0.406]
-              std: [0.229, 0.224, 0.225]
-              order: ''
+      name: "ImageNetDataset"
+      image_root: "./dataset/ILSVRC2012/"
+      cls_label_path: "./dataset/ILSVRC2012/train_list.txt"
+      transform_ops:
+        - RandCropImage:
+            size: 224
+        - RandFlipImage:
+            flip_code: 1
+        - NormalizeImage:
+            scale: 1.0/255.0
+            mean: [0.485, 0.456, 0.406]
+            std: [0.229, 0.224, 0.225]
+            order: ''
 
     sampler:
-        name: DistributedBatchSampler
-        batch_size: 64
-        drop_last: False
-        shuffle: True
+      name: "DistributedBatchSampler"
+      batch_size: 64
+      drop_last: False
+      shuffle: True
     loader:
-        num_workers: 6
-        use_shared_memory: True
+      num_workers: 4
+      use_shared_memory: True
 
   Eval:
     # TOTO: modify to the latest trainer
     dataset: 
-        name: ImageNetDataset
-        image_root: "./dataset/ILSVRC2012/"
-        cls_label_path: "./dataset/ILSVRC2012/val_list.txt"
-        transform_ops:
-          - ResizeImage:
-              resize_short: 256
-          - CropImage:
-              size: 224
-          - NormalizeImage:
-              scale: 0.00392157
-              mean: [0.485, 0.456, 0.406]
-              std: [0.229, 0.224, 0.225]
-              order: ''
+      name: "ImageNetDataset"
+      image_root: "./dataset/ILSVRC2012/"
+      cls_label_path: "./dataset/ILSVRC2012/val_list.txt"
+      transform_ops:
+        - ResizeImage:
+            resize_short: 256
+        - CropImage:
+            size: 224
+        - NormalizeImage:
+            scale: 1.0/255.0
+            mean: [0.485, 0.456, 0.406]
+            std: [0.229, 0.224, 0.225]
+            order: ''
     sampler:
-        name: DistributedBatchSampler
-        batch_size: 64
-        drop_last: False
-        shuffle: False
+      name: "DistributedBatchSampler"
+      batch_size: 64
+      drop_last: False
+      shuffle: False
     loader:
-        num_workers: 6
-        use_shared_memory: True
+      num_workers: 4
+      use_shared_memory: True
 
 Infer:
   infer_imgs: "docs/images/whl/demo.jpg"
   batch_size: 10
   transforms:
-      - DecodeImage:
-          to_rgb: True
-          channel_first: False
-      - ResizeImage:
-          resize_short: 256
-      - CropImage:
-          size: 224
-      - NormalizeImage:
-          scale: 1.0/255.0
-          mean: [0.485, 0.456, 0.406]
-          std: [0.229, 0.224, 0.225]
-          order: ''
-      - ToCHWImage:
+    - DecodeImage:
+        to_rgb: True
+        channel_first: False
+    - ResizeImage:
+        resize_short: 256
+    - CropImage:
+        size: 224
+    - NormalizeImage:
+        scale: 1.0/255.0
+        mean: [0.485, 0.456, 0.406]
+        std: [0.229, 0.224, 0.225]
+        order: ''
+    - ToCHWImage:
   PostProcess:
-    name: Topk
+    name: "Topk"
     topk: 5
     class_id_map_file: "ppcls/utils/imagenet1k_label_list.txt"
 
 Metric:
-    Train:
+  Train:
     - TopkAcc:
         topk: [1, 5]
-    Eval:
+  Eval:
     - TopkAcc:
         topk: [1, 5]

ppcls/configs/Products/ResNet50_vd_Aliproduct.yaml

@@ -21,7 +21,7 @@ Arch:
   name: "RecModel"
   Backbone: 
     name: "ResNet50_vd"
-    pretrained: False
+    pretrained: True
   BackboneStopLayer:
     name: "flatten_0"
   Neck:
@@ -56,52 +56,52 @@ Optimizer:
 DataLoader:
   Train:
     dataset:
-        name: "ImageNetDataset"
-        image_root: "./dataset/Aliproduct/"
-        cls_label_path: "./dataset/Aliproduct/train_list.txt"
-        transform_ops:
-          - ResizeImage:
-              size: 224
-          - RandFlipImage:
-              flip_code: 1
-          - NormalizeImage:
-              scale: 0.00392157
-              mean: [0.485, 0.456, 0.406]
-              std: [0.229, 0.224, 0.225]
-              order: ''
+      name: "ImageNetDataset"
+      image_root: "./dataset/Aliproduct/"
+      cls_label_path: "./dataset/Aliproduct/train_list.txt"
+      transform_ops:
+        - ResizeImage:
+            size: 224
+        - RandFlipImage:
+            flip_code: 1
+        - NormalizeImage:
+            scale: 0.00392157
+            mean: [0.485, 0.456, 0.406]
+            std: [0.229, 0.224, 0.225]
+            order: ''
 
     sampler:
-        name: DistributedBatchSampler
-        batch_size: 64
-        drop_last: False
-        shuffle: True
+      name: DistributedBatchSampler
+      batch_size: 64
+      drop_last: False
+      shuffle: True
     loader:
-        num_workers: 6
-        use_shared_memory: True
+      num_workers: 4
+      use_shared_memory: True
   Eval:
     # TOTO: modify to the latest trainer
     dataset: 
-        name: "ImageNetDataset"
-        image_root: "./dataset/Aliproduct/"
-        cls_label_path: "./dataset/Aliproduct/val_list.txt"
-        transform_ops:
-          - ResizeImage:
-              resize_short: 256
-          - CropImage:
-              size: 224
-          - NormalizeImage:
-              scale: 0.00392157
-              mean: [0.485, 0.456, 0.406]
-              std: [0.229, 0.224, 0.225]
-              order: ''
+      name: "ImageNetDataset"
+      image_root: "./dataset/Aliproduct/"
+      cls_label_path: "./dataset/Aliproduct/val_list.txt"
+      transform_ops:
+        - ResizeImage:
+            resize_short: 256
+        - CropImage:
+            size: 224
+        - NormalizeImage:
+            scale: 0.00392157
+            mean: [0.485, 0.456, 0.406]
+            std: [0.229, 0.224, 0.225]
+            order: ''
     sampler:
-        name: DistributedBatchSampler
-        batch_size: 64
-        drop_last: False
-        shuffle: False
+      name: DistributedBatchSampler
+      batch_size: 64
+      drop_last: False
+      shuffle: False
     loader:
-        num_workers: 6
-        use_shared_memory: True
+      num_workers: 4
+      use_shared_memory: True
 Metric:
   Train:
     - TopkAcc:
@@ -114,14 +114,14 @@ Infer:
   infer_imgs: "docs/images/whl/demo.jpg"
   batch_size: 10
   transforms:
-      - DecodeImage:
-          to_rgb: True
-          channel_first: False
-      - ResizeImage:
-          resize_short: 224
-      - NormalizeImage:
-          scale: 1.0/255.0
-          mean: [0.485, 0.456, 0.406]
-          std: [0.229, 0.224, 0.225]
-          order: ''
-      - ToCHWImage:
+    - DecodeImage:
+        to_rgb: True
+        channel_first: False
+    - ResizeImage:
+        resize_short: 224
+    - NormalizeImage:
+        scale: 1.0/255.0
+        mean: [0.485, 0.456, 0.406]
+        std: [0.229, 0.224, 0.225]
+        order: ''
+    - ToCHWImage:

ppcls/configs/Products/ResNet50_vd_Inshop.yaml

 # global configs
 Global:
   checkpoints: null
-  pretrained_model: null
+# please download pretrained model via this link:
+# https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/rec/models/pretrain/product_ResNet50_vd_Aliproduct_v1.0_pretrained.pdparams
+  pretrained_model: "product_ResNet50_vd_Aliproduct_v1.0_pretrained"
   output_dir: "./output/"
   device: "gpu"
   class_num: 3997
@@ -19,6 +21,9 @@ Global:
 # model architecture
 Arch:
   name: "RecModel"
+  infer_output_key: "features"
+  infer_add_softmax: False
+  
   Backbone: 
     name: "ResNet50_vd"
     pretrained: False
@@ -34,8 +39,6 @@ Arch:
     class_num: 3997
     margin: 0.15
     scale: 30
-  infer_output_key: "features"
-  infer_add_softmax: False
    
 # loss function config for traing/eval process
 Loss:
@@ -67,38 +70,37 @@ Optimizer:
 DataLoader:
   Train:
     dataset:
-        name: "ImageNetDataset"
-        image_root: "./dataset/Inshop/"
-        cls_label_path: "./dataset/Inshop/train_list.txt"
-        transform_ops:
-          - ResizeImage:
-              size: 224
-          - RandFlipImage:
-              flip_code: 1
-          - NormalizeImage:
-              scale: 0.00392157
-              mean: [0.485, 0.456, 0.406]
-              std: [0.229, 0.224, 0.225]
-              order: ''
-          - RandomErasing:
-              EPSILON: 0.5
-              sl: 0.02
-              sh: 0.4
-              r1: 0.3
-              mean: [0., 0., 0.]
-
+      name: "ImageNetDataset"
+      image_root: "./dataset/Inshop/"
+      cls_label_path: "./dataset/Inshop/train_list.txt"
+      transform_ops:
+        - ResizeImage:
+            size: 224
+        - RandFlipImage:
+            flip_code: 1
+        - NormalizeImage:
+            scale: 0.00392157
+            mean: [0.485, 0.456, 0.406]
+            std: [0.229, 0.224, 0.225]
+            order: ''
+        - RandomErasing:
+            EPSILON: 0.5
+            sl: 0.02
+            sh: 0.4
+            r1: 0.3
+            mean: [0., 0., 0.]
     sampler:
-        name: DistributedRandomIdentitySampler
-        batch_size: 64
-        num_instances: 2
-        drop_last: False
-        shuffle: True
+      name: "DistributedRandomIdentitySampler"
+      batch_size: 64
+      num_instances: 2
+      drop_last: False
+      shuffle: True
     loader:
-        num_workers: 6
-        use_shared_memory: True
+      num_workers: 4
+      use_shared_memory: True
+      
   Eval:
     Query:
-    # TOTO: modify to the latest trainer
       dataset: 
         name: "ImageNetDataset"
         image_root: "./dataset/Inshop/"
@@ -112,16 +114,15 @@ DataLoader:
               std: [0.229, 0.224, 0.225]
               order: ''
       sampler:
-        name: DistributedBatchSampler
+        name: "DistributedBatchSampler"
         batch_size: 64
         drop_last: False
         shuffle: False
       loader:
-        num_workers: 6
+        num_workers: 4
         use_shared_memory: True
 
     Gallery:
-    # TOTO: modify to the latest trainer
       dataset: 
         name: "ImageNetDataset"
         image_root: "./dataset/Inshop/"
@@ -130,17 +131,17 @@ DataLoader:
           - ResizeImage:
               size: 224
           - NormalizeImage:
-              scale: 0.00392157
+              scale: 1.0/255.0
               mean: [0.485, 0.456, 0.406]
               std: [0.229, 0.224, 0.225]
               order: ''
       sampler:
-        name: DistributedBatchSampler
+        name: "DistributedBatchSampler"
         batch_size: 64
         drop_last: False
         shuffle: False
       loader:
-        num_workers: 6
+        num_workers: 4
         use_shared_memory: True
 
 Metric:
@@ -152,14 +153,14 @@ Infer:
   infer_imgs: "docs/images/whl/demo.jpg"
   batch_size: 10
   transforms:
-      - DecodeImage:
-          to_rgb: True
-          channel_first: False
-      - ResizeImage:
-          resize_short: 224
-      - NormalizeImage:
-          scale: 1.0/255.0
-          mean: [0.485, 0.456, 0.406]
-          std: [0.229, 0.224, 0.225]
-          order: ''
-      - ToCHWImage:
+    - DecodeImage:
+        to_rgb: True
+        channel_first: False
+    - ResizeImage:
+        resize_short: 224
+    - NormalizeImage:
+        scale: 1.0/255.0
+        mean: [0.485, 0.456, 0.406]
+        std: [0.229, 0.224, 0.225]
+        order: ''
+    - ToCHWImage:

ppcls/configs/Products/ResNet50_vd_SOP.yaml

 # global configs
 Global:
   checkpoints: null
-  pretrained_model: null
+# please download pretrained model via this link:
+# https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/rec/models/pretrain/product_ResNet50_vd_Aliproduct_v1.0_pretrained.pdparams
+  pretrained_model: "product_ResNet50_vd_Aliproduct_v1.0_pretrained"
   output_dir: "./output/"
   device: "gpu"
   class_num: 11319
@@ -67,38 +69,37 @@ Optimizer:
 DataLoader:
   Train:
     dataset:
-        name: "ImageNetDataset"
-        image_root: "./dataset/Stanford_Online_Products/"
-        cls_label_path: "./dataset/Stanford_Online_Products/train_list.txt"
-        transform_ops:
-          - ResizeImage:
-              size: 224
-          - RandFlipImage:
-              flip_code: 1
-          - NormalizeImage:
-              scale: 0.00392157
-              mean: [0.485, 0.456, 0.406]
-              std: [0.229, 0.224, 0.225]
-              order: ''
-          - RandomErasing:
-              EPSILON: 0.5
-              sl: 0.02
-              sh: 0.4
-              r1: 0.3
-              mean: [0., 0., 0.]
+      name: "ImageNetDataset"
+      image_root: "./dataset/Stanford_Online_Products/"
+      cls_label_path: "./dataset/Stanford_Online_Products/train_list.txt"
+      transform_ops:
+        - ResizeImage:
+            size: 224
+        - RandFlipImage:
+            flip_code: 1
+        - NormalizeImage:
+            scale: 0.00392157
+            mean: [0.485, 0.456, 0.406]
+            std: [0.229, 0.224, 0.225]
+            order: ''
+        - RandomErasing:
+            EPSILON: 0.5
+            sl: 0.02
+            sh: 0.4
+            r1: 0.3
+            mean: [0., 0., 0.]
 
     sampler:
-        name: DistributedRandomIdentitySampler
-        batch_size: 64
-        num_instances: 2
-        drop_last: False
-        shuffle: True
+      name: DistributedRandomIdentitySampler
+      batch_size: 64
+      num_instances: 2
+      drop_last: False
+      shuffle: True
     loader:
-        num_workers: 6
-        use_shared_memory: True
+      num_workers: 6
+      use_shared_memory: True
   Eval:
     Query:
-    # TOTO: modify to the latest trainer
       dataset: 
         name: "ImageNetDataset"
         image_root: "./dataset/Stanford_Online_Products/"
@@ -117,11 +118,10 @@ DataLoader:
         drop_last: False
         shuffle: False
       loader:
-        num_workers: 6
+        num_workers: 4
         use_shared_memory: True
 
     Gallery:
-    # TOTO: modify to the latest trainer
       dataset: 
         name: "ImageNetDataset"
         image_root: "./dataset/Stanford_Online_Products/"
@@ -140,7 +140,7 @@ DataLoader:
         drop_last: False
         shuffle: False
       loader:
-        num_workers: 6
+        num_workers: 4
         use_shared_memory: True
 
 Metric:
@@ -152,14 +152,14 @@ Infer:
   infer_imgs: "docs/images/whl/demo.jpg"
   batch_size: 10
   transforms:
-      - DecodeImage:
-          to_rgb: True
-          channel_first: False
-      - ResizeImage:
-          resize_short: 224
-      - NormalizeImage:
-          scale: 1.0/255.0
-          mean: [0.485, 0.456, 0.406]
-          std: [0.229, 0.224, 0.225]
-          order: ''
-      - ToCHWImage:
+    - DecodeImage:
+        to_rgb: True
+        channel_first: False
+    - ResizeImage:
+        resize_short: 224
+    - NormalizeImage:
+        scale: 1.0/255.0
+        mean: [0.485, 0.456, 0.406]
+        std: [0.229, 0.224, 0.225]
+        order: ''
+    - ToCHWImage: