complete amp args

ppcls/configs/ImageNet/Distillation/PPLCNet_x2_5_dml.yaml

@@ -15,8 +15,11 @@ Global:
   save_inference_dir: ./inference
 
 AMP:
+  use_amp: True
+  use_fp16_test: False
   scale_loss: 128.0
   use_dynamic_loss_scaling: True
+  use_promote: False
   # O1: mixed fp16, O2: pure fp16
   level: O1
 

ppcls/configs/ImageNet/Distillation/PPLCNet_x2_5_ssld.yaml

@@ -16,8 +16,11 @@ Global:
   to_static: True
 
 AMP:
+  use_amp: True
+  use_fp16_test: False
   scale_loss: 128.0
   use_dynamic_loss_scaling: True
+  use_promote: False
   # O1: mixed fp16, O2: pure fp16
   level: O1
 

ppcls/configs/ImageNet/Distillation/PPLCNet_x2_5_udml.yaml

@@ -15,8 +15,11 @@ Global:
   save_inference_dir: ./inference
 
 AMP:
+  use_amp: True
+  use_fp16_test: False
   scale_loss: 128.0
   use_dynamic_loss_scaling: True
+  use_promote: False
   # O1: mixed fp16, O2: pure fp16
   level: O1
 

ppcls/configs/ImageNet/Distillation/res2net200_vd_distill_pphgnet_base.yaml

@@ -17,8 +17,11 @@ Global:
 
 # mixed precision training
 AMP:
+  use_amp: True
+  use_fp16_test: False
   scale_loss: 128.0
   use_dynamic_loss_scaling: True
+  use_promote: False
   # O1: mixed fp16, O2: pure fp16
   level: O1
 

ppcls/configs/ImageNet/EfficientNetV2/EfficientNetV2_S.yaml

@@ -16,8 +16,11 @@ Global:
   save_inference_dir: ./inference
 
 AMP:
+  use_amp: True
+  use_fp16_test: False
   scale_loss: 65536
   use_dynamic_loss_scaling: True
+  use_promote: False
   # O1: mixed fp16, O2: pure fp16
   level: O1
 

ppcls/configs/ImageNet/MobileNetV3/MobileNetV3_small_x1_0_ampo2_ultra.yaml

@@ -17,8 +17,11 @@ Global:
 
 # mixed precision training
 AMP:
+  use_amp: True
+  use_fp16_test: False
   scale_loss: 128.0
   use_dynamic_loss_scaling: True
+  use_promote: False
   # O1: mixed fp16, O2: pure fp16
   level: O2
 

ppcls/configs/ImageNet/MobileViTV2/MobileViTV2_x0_5.yaml

@@ -17,8 +17,11 @@ Global:
 
 # mixed precision training
 AMP:
+  use_amp: True
+  use_fp16_test: False
   scale_loss: 65536
   use_dynamic_loss_scaling: True
+  use_promote: False
   # O1: mixed fp16, O2: pure fp16
   level: O1
 

ppcls/configs/ImageNet/MobileViTV2/MobileViTV2_x1_0.yaml

@@ -17,8 +17,11 @@ Global:
 
 # mixed precision training
 AMP:
+  use_amp: True
+  use_fp16_test: False
   scale_loss: 65536
   use_dynamic_loss_scaling: True
+  use_promote: False
   # O1: mixed fp16, O2: pure fp16
   level: O1
 

ppcls/configs/ImageNet/MobileViTV2/MobileViTV2_x1_5.yaml

@@ -17,8 +17,11 @@ Global:
 
 # mixed precision training
 AMP:
+  use_amp: True
+  use_fp16_test: False
   scale_loss: 65536
   use_dynamic_loss_scaling: True
+  use_promote: False
   # O1: mixed fp16, O2: pure fp16
   level: O1
 

ppcls/configs/ImageNet/MobileViTV2/MobileViTV2_x2_0.yaml

@@ -17,8 +17,11 @@ Global:
 
 # mixed precision training
 AMP:
+  use_amp: True
+  use_fp16_test: False
   scale_loss: 65536
   use_dynamic_loss_scaling: True
+  use_promote: False
   # O1: mixed fp16, O2: pure fp16
   level: O1
 

ppcls/configs/ImageNet/MobileViTV3/MobileViTV3_S.yaml

@@ -17,8 +17,11 @@ Global:
 
 # mixed precision training
 AMP:
+  use_amp: True
+  use_fp16_test: False
   scale_loss: 65536
   use_dynamic_loss_scaling: True
+  use_promote: False
   # O1: mixed fp16, O2: pure fp16
   level: O1
 

ppcls/configs/ImageNet/MobileViTV3/MobileViTV3_S_L2.yaml

@@ -17,8 +17,11 @@ Global:
 
 # mixed precision training
 AMP:
+  use_amp: True
+  use_fp16_test: False
   scale_loss: 65536
   use_dynamic_loss_scaling: True
+  use_promote: False
   # O1: mixed fp16, O2: pure fp16
   level: O1
 

ppcls/configs/ImageNet/MobileViTV3/MobileViTV3_XS.yaml

@@ -17,8 +17,11 @@ Global:
 
 # mixed precision training
 AMP:
+  use_amp: True
+  use_fp16_test: False
   scale_loss: 65536
   use_dynamic_loss_scaling: True
+  use_promote: False
   # O1: mixed fp16, O2: pure fp16
   level: O1
 

ppcls/configs/ImageNet/MobileViTV3/MobileViTV3_XS_L2.yaml

@@ -17,8 +17,11 @@ Global:
 
 # mixed precision training
 AMP:
+  use_amp: True
+  use_fp16_test: False
   scale_loss: 65536
   use_dynamic_loss_scaling: True
+  use_promote: False
   # O1: mixed fp16, O2: pure fp16
   level: O1
 

ppcls/configs/ImageNet/MobileViTV3/MobileViTV3_XXS.yaml

@@ -17,8 +17,11 @@ Global:
 
 # mixed precision training
 AMP:
+  use_amp: True
+  use_fp16_test: False
   scale_loss: 65536
   use_dynamic_loss_scaling: True
+  use_promote: False
   # O1: mixed fp16, O2: pure fp16
   level: O1
 

ppcls/configs/ImageNet/MobileViTV3/MobileViTV3_XXS_L2.yaml

@@ -17,8 +17,11 @@ Global:
 
 # mixed precision training
 AMP:
+  use_amp: True
+  use_fp16_test: False
   scale_loss: 65536
   use_dynamic_loss_scaling: True
+  use_promote: False
   # O1: mixed fp16, O2: pure fp16
   level: O1
 

ppcls/configs/ImageNet/MobileViTV3/MobileViTV3_x0_5.yaml

@@ -17,8 +17,11 @@ Global:
 
 # mixed precision training
 AMP:
+  use_amp: True
+  use_fp16_test: False
   scale_loss: 65536
   use_dynamic_loss_scaling: True
+  use_promote: False
   # O1: mixed fp16, O2: pure fp16
   level: O1
 

ppcls/configs/ImageNet/MobileViTV3/MobileViTV3_x0_75.yaml

@@ -17,8 +17,11 @@ Global:
 
 # mixed precision training
 AMP:
+  use_amp: True
+  use_fp16_test: False
   scale_loss: 65536
   use_dynamic_loss_scaling: True
+  use_promote: False
   # O1: mixed fp16, O2: pure fp16
   level: O1
 

ppcls/configs/ImageNet/MobileViTV3/MobileViTV3_x1_0.yaml

@@ -17,8 +17,11 @@ Global:
 
 # mixed precision training
 AMP:
+  use_amp: True
+  use_fp16_test: False
   scale_loss: 65536
   use_dynamic_loss_scaling: True
+  use_promote: False
   # O1: mixed fp16, O2: pure fp16
   level: O1
 

ppcls/configs/ImageNet/NextViT/NextViT_base_224.yaml

@@ -19,8 +19,11 @@ Global:
 
 # mixed precision training
 AMP:
+  use_amp: True
+  use_fp16_test: False
   scale_loss: 128.0
   use_dynamic_loss_scaling: True
+  use_promote: False
   # O1: mixed fp16, O2: pure fp16
   level: O1
 

ppcls/configs/ImageNet/NextViT/NextViT_base_384.yaml

@@ -19,8 +19,11 @@ Global:
 
 # mixed precision training
 AMP:
+  use_amp: True
+  use_fp16_test: False
   scale_loss: 128.0
   use_dynamic_loss_scaling: True
+  use_promote: False
   # O1: mixed fp16, O2: pure fp16
   level: O1
 

ppcls/configs/ImageNet/NextViT/NextViT_large_224.yaml

@@ -19,8 +19,11 @@ Global:
 
 # mixed precision training
 AMP:
+  use_amp: True
+  use_fp16_test: False
   scale_loss: 128.0
   use_dynamic_loss_scaling: True
+  use_promote: False
   # O1: mixed fp16, O2: pure fp16
   level: O1
 

ppcls/configs/ImageNet/NextViT/NextViT_large_384.yaml

@@ -19,8 +19,11 @@ Global:
 
 # mixed precision training
 AMP:
+  use_amp: True
+  use_fp16_test: False
   scale_loss: 128.0
   use_dynamic_loss_scaling: True
+  use_promote: False
   # O1: mixed fp16, O2: pure fp16
   level: O1
 

ppcls/configs/ImageNet/NextViT/NextViT_small_224.yaml

@@ -19,8 +19,11 @@ Global:
 
 # mixed precision training
 AMP:
+  use_amp: True
+  use_fp16_test: False
   scale_loss: 128.0
   use_dynamic_loss_scaling: True
+  use_promote: False
   # O1: mixed fp16, O2: pure fp16
   level: O1
 

ppcls/configs/ImageNet/NextViT/NextViT_small_384.yaml

@@ -19,8 +19,11 @@ Global:
 
 # mixed precision training
 AMP:
+  use_amp: True
+  use_fp16_test: False
   scale_loss: 128.0
   use_dynamic_loss_scaling: True
+  use_promote: False
   # O1: mixed fp16, O2: pure fp16
   level: O1
 

ppcls/configs/ImageNet/PPHGNet/PPHGNet_base.yaml

@@ -19,8 +19,11 @@ Global:
 
 # mixed precision training
 AMP:
+  use_amp: True
+  use_fp16_test: False
   scale_loss: 128.0
   use_dynamic_loss_scaling: True
+  use_promote: False
   # O1: mixed fp16, O2: pure fp16
   level: O1
 

ppcls/configs/ImageNet/PPHGNet/PPHGNet_small.yaml

@@ -19,8 +19,11 @@ Global:
 
 # mixed precision training
 AMP:
+  use_amp: True
+  use_fp16_test: False
   scale_loss: 128.0
   use_dynamic_loss_scaling: True
+  use_promote: False
   # O1: mixed fp16, O2: pure fp16
   level: O1
 

ppcls/configs/ImageNet/PPHGNet/PPHGNet_tiny.yaml

@@ -19,8 +19,11 @@ Global:
 
 # mixed precision training
 AMP:
+  use_amp: True
+  use_fp16_test: False
   scale_loss: 128.0
   use_dynamic_loss_scaling: True
+  use_promote: False
   # O1: mixed fp16, O2: pure fp16
   level: O1
 

ppcls/configs/ImageNet/PPLCNet/PPLCNet_x1_0_ampo2_ultra.yaml

@@ -17,8 +17,11 @@ Global:
 
 # mixed precision training
 AMP:
+  use_amp: True
+  use_fp16_test: False
   scale_loss: 128.0
   use_dynamic_loss_scaling: True
+  use_promote: False
   # O1: mixed fp16, O2: pure fp16
   level: O2
 

ppcls/configs/ImageNet/ResNet/ResNet50_amp_O1.yaml

@@ -19,8 +19,11 @@ Global:
 
 # mixed precision training
 AMP:
+  use_amp: True
+  use_fp16_test: False
   scale_loss: 128.0
   use_dynamic_loss_scaling: True
+  use_promote: False
   # O1: mixed fp16, O2: pure fp16
   level: O1
 

ppcls/configs/ImageNet/ResNet/ResNet50_amp_O1_ultra.yaml

@@ -20,8 +20,11 @@ Global:
 
 # mixed precision training
 AMP:
+  use_amp: True
+  use_fp16_test: False
   scale_loss: 128.0
   use_dynamic_loss_scaling: True
+  use_promote: False
   # O1: mixed fp16, O2: pure fp16
   level: O1
 

ppcls/configs/ImageNet/ResNet/ResNet50_amp_O2_ultra.yaml

@@ -20,8 +20,11 @@ Global:
 
 # mixed precision training
 AMP:
+  use_amp: True
+  use_fp16_test: False
   scale_loss: 128.0
   use_dynamic_loss_scaling: True
+  use_promote: False
   # O1: mixed fp16, O2: pure fp16
   level: O2
 

ppcls/configs/ImageNet/ResNet/ResNet50_ampo2_ultra.yaml

@@ -19,8 +19,11 @@ Global:
 
 # mixed precision training
 AMP:
+  use_amp: True
+  use_fp16_test: False
   scale_loss: 128.0
   use_dynamic_loss_scaling: True
+  use_promote: False
   # O1: mixed fp16, O2: pure fp16
   level: O2
 

ppcls/configs/ImageNet/SENet/SE_ResNeXt101_32x4d_amp_O2_ultra.yaml

@@ -17,8 +17,11 @@ Global:
 
 # mixed precision training
 AMP:
+  use_amp: True
+  use_fp16_test: False
   scale_loss: 128.0
   use_dynamic_loss_scaling: True
+  use_promote: False
   # O1: mixed fp16, O2: pure fp16
   level: O2