From 09817fe8595cb240438a5d630443f71ccfa5cc3c Mon Sep 17 00:00:00 2001 From: gaotingquan Date: Mon, 29 May 2023 09:14:57 +0000 Subject: [PATCH] complete amp args --- ppcls/configs/ImageNet/Distillation/PPLCNet_x2_5_dml.yaml | 3 +++ ppcls/configs/ImageNet/Distillation/PPLCNet_x2_5_ssld.yaml | 3 +++ ppcls/configs/ImageNet/Distillation/PPLCNet_x2_5_udml.yaml | 3 +++ .../Distillation/res2net200_vd_distill_pphgnet_base.yaml | 3 +++ ppcls/configs/ImageNet/EfficientNetV2/EfficientNetV2_S.yaml | 3 +++ .../MobileNetV3/MobileNetV3_small_x1_0_ampo2_ultra.yaml | 3 +++ ppcls/configs/ImageNet/MobileViTV2/MobileViTV2_x0_5.yaml | 3 +++ ppcls/configs/ImageNet/MobileViTV2/MobileViTV2_x1_0.yaml | 3 +++ ppcls/configs/ImageNet/MobileViTV2/MobileViTV2_x1_5.yaml | 3 +++ ppcls/configs/ImageNet/MobileViTV2/MobileViTV2_x2_0.yaml | 3 +++ ppcls/configs/ImageNet/MobileViTV3/MobileViTV3_S.yaml | 3 +++ ppcls/configs/ImageNet/MobileViTV3/MobileViTV3_S_L2.yaml | 3 +++ ppcls/configs/ImageNet/MobileViTV3/MobileViTV3_XS.yaml | 3 +++ ppcls/configs/ImageNet/MobileViTV3/MobileViTV3_XS_L2.yaml | 3 +++ ppcls/configs/ImageNet/MobileViTV3/MobileViTV3_XXS.yaml | 3 +++ ppcls/configs/ImageNet/MobileViTV3/MobileViTV3_XXS_L2.yaml | 3 +++ ppcls/configs/ImageNet/MobileViTV3/MobileViTV3_x0_5.yaml | 3 +++ ppcls/configs/ImageNet/MobileViTV3/MobileViTV3_x0_75.yaml | 3 +++ ppcls/configs/ImageNet/MobileViTV3/MobileViTV3_x1_0.yaml | 3 +++ ppcls/configs/ImageNet/NextViT/NextViT_base_224.yaml | 3 +++ ppcls/configs/ImageNet/NextViT/NextViT_base_384.yaml | 3 +++ ppcls/configs/ImageNet/NextViT/NextViT_large_224.yaml | 3 +++ ppcls/configs/ImageNet/NextViT/NextViT_large_384.yaml | 3 +++ ppcls/configs/ImageNet/NextViT/NextViT_small_224.yaml | 3 +++ ppcls/configs/ImageNet/NextViT/NextViT_small_384.yaml | 3 +++ ppcls/configs/ImageNet/PPHGNet/PPHGNet_base.yaml | 3 +++ ppcls/configs/ImageNet/PPHGNet/PPHGNet_small.yaml | 3 +++ ppcls/configs/ImageNet/PPHGNet/PPHGNet_tiny.yaml | 3 +++ ppcls/configs/ImageNet/PPLCNet/PPLCNet_x1_0_ampo2_ultra.yaml | 3 +++ ppcls/configs/ImageNet/ResNet/ResNet50_amp_O1.yaml | 3 +++ ppcls/configs/ImageNet/ResNet/ResNet50_amp_O1_ultra.yaml | 3 +++ ppcls/configs/ImageNet/ResNet/ResNet50_amp_O2_ultra.yaml | 3 +++ ppcls/configs/ImageNet/ResNet/ResNet50_ampo2_ultra.yaml | 3 +++ .../ImageNet/SENet/SE_ResNeXt101_32x4d_amp_O2_ultra.yaml | 3 +++ 34 files changed, 102 insertions(+) diff --git a/ppcls/configs/ImageNet/Distillation/PPLCNet_x2_5_dml.yaml b/ppcls/configs/ImageNet/Distillation/PPLCNet_x2_5_dml.yaml index 494fa7f9..4fb76809 100644 --- a/ppcls/configs/ImageNet/Distillation/PPLCNet_x2_5_dml.yaml +++ b/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 diff --git a/ppcls/configs/ImageNet/Distillation/PPLCNet_x2_5_ssld.yaml b/ppcls/configs/ImageNet/Distillation/PPLCNet_x2_5_ssld.yaml index c581985b..e778cb8d 100644 --- a/ppcls/configs/ImageNet/Distillation/PPLCNet_x2_5_ssld.yaml +++ b/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 diff --git a/ppcls/configs/ImageNet/Distillation/PPLCNet_x2_5_udml.yaml b/ppcls/configs/ImageNet/Distillation/PPLCNet_x2_5_udml.yaml index faebc855..17e020e9 100644 --- a/ppcls/configs/ImageNet/Distillation/PPLCNet_x2_5_udml.yaml +++ b/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 diff --git a/ppcls/configs/ImageNet/Distillation/res2net200_vd_distill_pphgnet_base.yaml b/ppcls/configs/ImageNet/Distillation/res2net200_vd_distill_pphgnet_base.yaml index ccd4fdf2..acc0aa38 100644 --- a/ppcls/configs/ImageNet/Distillation/res2net200_vd_distill_pphgnet_base.yaml +++ b/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 diff --git a/ppcls/configs/ImageNet/EfficientNetV2/EfficientNetV2_S.yaml b/ppcls/configs/ImageNet/EfficientNetV2/EfficientNetV2_S.yaml index 05eaadc2..48257bc1 100644 --- a/ppcls/configs/ImageNet/EfficientNetV2/EfficientNetV2_S.yaml +++ b/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 diff --git a/ppcls/configs/ImageNet/MobileNetV3/MobileNetV3_small_x1_0_ampo2_ultra.yaml b/ppcls/configs/ImageNet/MobileNetV3/MobileNetV3_small_x1_0_ampo2_ultra.yaml index 9b7f81d9..2880583b 100644 --- a/ppcls/configs/ImageNet/MobileNetV3/MobileNetV3_small_x1_0_ampo2_ultra.yaml +++ b/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 diff --git a/ppcls/configs/ImageNet/MobileViTV2/MobileViTV2_x0_5.yaml b/ppcls/configs/ImageNet/MobileViTV2/MobileViTV2_x0_5.yaml index 87f9d187..12891491 100644 --- a/ppcls/configs/ImageNet/MobileViTV2/MobileViTV2_x0_5.yaml +++ b/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 diff --git a/ppcls/configs/ImageNet/MobileViTV2/MobileViTV2_x1_0.yaml b/ppcls/configs/ImageNet/MobileViTV2/MobileViTV2_x1_0.yaml index a9101b9a..aeee054d 100644 --- a/ppcls/configs/ImageNet/MobileViTV2/MobileViTV2_x1_0.yaml +++ b/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 diff --git a/ppcls/configs/ImageNet/MobileViTV2/MobileViTV2_x1_5.yaml b/ppcls/configs/ImageNet/MobileViTV2/MobileViTV2_x1_5.yaml index 4201758d..198e4333 100644 --- a/ppcls/configs/ImageNet/MobileViTV2/MobileViTV2_x1_5.yaml +++ b/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 diff --git a/ppcls/configs/ImageNet/MobileViTV2/MobileViTV2_x2_0.yaml b/ppcls/configs/ImageNet/MobileViTV2/MobileViTV2_x2_0.yaml index aa28a5db..2739ea9d 100644 --- a/ppcls/configs/ImageNet/MobileViTV2/MobileViTV2_x2_0.yaml +++ b/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 diff --git a/ppcls/configs/ImageNet/MobileViTV3/MobileViTV3_S.yaml b/ppcls/configs/ImageNet/MobileViTV3/MobileViTV3_S.yaml index 14ab95a7..ba49dbbc 100644 --- a/ppcls/configs/ImageNet/MobileViTV3/MobileViTV3_S.yaml +++ b/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 diff --git a/ppcls/configs/ImageNet/MobileViTV3/MobileViTV3_S_L2.yaml b/ppcls/configs/ImageNet/MobileViTV3/MobileViTV3_S_L2.yaml index c2fdf571..4a9ba8b0 100644 --- a/ppcls/configs/ImageNet/MobileViTV3/MobileViTV3_S_L2.yaml +++ b/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 diff --git a/ppcls/configs/ImageNet/MobileViTV3/MobileViTV3_XS.yaml b/ppcls/configs/ImageNet/MobileViTV3/MobileViTV3_XS.yaml index e3b9ab4b..3ac7264a 100644 --- a/ppcls/configs/ImageNet/MobileViTV3/MobileViTV3_XS.yaml +++ b/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 diff --git a/ppcls/configs/ImageNet/MobileViTV3/MobileViTV3_XS_L2.yaml b/ppcls/configs/ImageNet/MobileViTV3/MobileViTV3_XS_L2.yaml index e22c32dc..10b3c7f9 100644 --- a/ppcls/configs/ImageNet/MobileViTV3/MobileViTV3_XS_L2.yaml +++ b/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 diff --git a/ppcls/configs/ImageNet/MobileViTV3/MobileViTV3_XXS.yaml b/ppcls/configs/ImageNet/MobileViTV3/MobileViTV3_XXS.yaml index a0bd2e36..719a2006 100644 --- a/ppcls/configs/ImageNet/MobileViTV3/MobileViTV3_XXS.yaml +++ b/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 diff --git a/ppcls/configs/ImageNet/MobileViTV3/MobileViTV3_XXS_L2.yaml b/ppcls/configs/ImageNet/MobileViTV3/MobileViTV3_XXS_L2.yaml index 29991159..ff84fadd 100644 --- a/ppcls/configs/ImageNet/MobileViTV3/MobileViTV3_XXS_L2.yaml +++ b/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 diff --git a/ppcls/configs/ImageNet/MobileViTV3/MobileViTV3_x0_5.yaml b/ppcls/configs/ImageNet/MobileViTV3/MobileViTV3_x0_5.yaml index 9efae98b..31eef0c3 100644 --- a/ppcls/configs/ImageNet/MobileViTV3/MobileViTV3_x0_5.yaml +++ b/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 diff --git a/ppcls/configs/ImageNet/MobileViTV3/MobileViTV3_x0_75.yaml b/ppcls/configs/ImageNet/MobileViTV3/MobileViTV3_x0_75.yaml index f262fda9..eb4e3a3b 100644 --- a/ppcls/configs/ImageNet/MobileViTV3/MobileViTV3_x0_75.yaml +++ b/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 diff --git a/ppcls/configs/ImageNet/MobileViTV3/MobileViTV3_x1_0.yaml b/ppcls/configs/ImageNet/MobileViTV3/MobileViTV3_x1_0.yaml index 3d24e00b..e6b437bb 100644 --- a/ppcls/configs/ImageNet/MobileViTV3/MobileViTV3_x1_0.yaml +++ b/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 diff --git a/ppcls/configs/ImageNet/NextViT/NextViT_base_224.yaml b/ppcls/configs/ImageNet/NextViT/NextViT_base_224.yaml index 7f334b31..66317732 100644 --- a/ppcls/configs/ImageNet/NextViT/NextViT_base_224.yaml +++ b/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 diff --git a/ppcls/configs/ImageNet/NextViT/NextViT_base_384.yaml b/ppcls/configs/ImageNet/NextViT/NextViT_base_384.yaml index 1bf91e59..d013e647 100644 --- a/ppcls/configs/ImageNet/NextViT/NextViT_base_384.yaml +++ b/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 diff --git a/ppcls/configs/ImageNet/NextViT/NextViT_large_224.yaml b/ppcls/configs/ImageNet/NextViT/NextViT_large_224.yaml index 07d64ed4..33b8dabf 100644 --- a/ppcls/configs/ImageNet/NextViT/NextViT_large_224.yaml +++ b/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 diff --git a/ppcls/configs/ImageNet/NextViT/NextViT_large_384.yaml b/ppcls/configs/ImageNet/NextViT/NextViT_large_384.yaml index 3881a468..a3eaf4c5 100644 --- a/ppcls/configs/ImageNet/NextViT/NextViT_large_384.yaml +++ b/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 diff --git a/ppcls/configs/ImageNet/NextViT/NextViT_small_224.yaml b/ppcls/configs/ImageNet/NextViT/NextViT_small_224.yaml index 60ffdff4..d16151df 100644 --- a/ppcls/configs/ImageNet/NextViT/NextViT_small_224.yaml +++ b/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 diff --git a/ppcls/configs/ImageNet/NextViT/NextViT_small_384.yaml b/ppcls/configs/ImageNet/NextViT/NextViT_small_384.yaml index 0f06f369..daf72762 100644 --- a/ppcls/configs/ImageNet/NextViT/NextViT_small_384.yaml +++ b/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 diff --git a/ppcls/configs/ImageNet/PPHGNet/PPHGNet_base.yaml b/ppcls/configs/ImageNet/PPHGNet/PPHGNet_base.yaml index ffccaf8c..873b027a 100644 --- a/ppcls/configs/ImageNet/PPHGNet/PPHGNet_base.yaml +++ b/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 diff --git a/ppcls/configs/ImageNet/PPHGNet/PPHGNet_small.yaml b/ppcls/configs/ImageNet/PPHGNet/PPHGNet_small.yaml index 5dad03d9..7e065cd4 100644 --- a/ppcls/configs/ImageNet/PPHGNet/PPHGNet_small.yaml +++ b/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 diff --git a/ppcls/configs/ImageNet/PPHGNet/PPHGNet_tiny.yaml b/ppcls/configs/ImageNet/PPHGNet/PPHGNet_tiny.yaml index 454f2815..e3fad8ff 100644 --- a/ppcls/configs/ImageNet/PPHGNet/PPHGNet_tiny.yaml +++ b/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 diff --git a/ppcls/configs/ImageNet/PPLCNet/PPLCNet_x1_0_ampo2_ultra.yaml b/ppcls/configs/ImageNet/PPLCNet/PPLCNet_x1_0_ampo2_ultra.yaml index 63c52c0a..336fb334 100644 --- a/ppcls/configs/ImageNet/PPLCNet/PPLCNet_x1_0_ampo2_ultra.yaml +++ b/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 diff --git a/ppcls/configs/ImageNet/ResNet/ResNet50_amp_O1.yaml b/ppcls/configs/ImageNet/ResNet/ResNet50_amp_O1.yaml index b8976e33..4981efe9 100644 --- a/ppcls/configs/ImageNet/ResNet/ResNet50_amp_O1.yaml +++ b/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 diff --git a/ppcls/configs/ImageNet/ResNet/ResNet50_amp_O1_ultra.yaml b/ppcls/configs/ImageNet/ResNet/ResNet50_amp_O1_ultra.yaml index f1ae9ee7..47aa4c0e 100644 --- a/ppcls/configs/ImageNet/ResNet/ResNet50_amp_O1_ultra.yaml +++ b/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 diff --git a/ppcls/configs/ImageNet/ResNet/ResNet50_amp_O2_ultra.yaml b/ppcls/configs/ImageNet/ResNet/ResNet50_amp_O2_ultra.yaml index ec414f2e..7f4de664 100644 --- a/ppcls/configs/ImageNet/ResNet/ResNet50_amp_O2_ultra.yaml +++ b/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 diff --git a/ppcls/configs/ImageNet/ResNet/ResNet50_ampo2_ultra.yaml b/ppcls/configs/ImageNet/ResNet/ResNet50_ampo2_ultra.yaml index 889bccf5..ee606707 100644 --- a/ppcls/configs/ImageNet/ResNet/ResNet50_ampo2_ultra.yaml +++ b/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 diff --git a/ppcls/configs/ImageNet/SENet/SE_ResNeXt101_32x4d_amp_O2_ultra.yaml b/ppcls/configs/ImageNet/SENet/SE_ResNeXt101_32x4d_amp_O2_ultra.yaml index 3eb06dfe..83578218 100644 --- a/ppcls/configs/ImageNet/SENet/SE_ResNeXt101_32x4d_amp_O2_ultra.yaml +++ b/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 -- GitLab