diff --git a/ppcls/configs/ImageNet/Distillation/PPLCNet_x2_5_dml.yaml b/ppcls/configs/ImageNet/Distillation/PPLCNet_x2_5_dml.yaml index 494fa7f973698cdf7e64ada31f56081e6e08a916..4fb7680988c668d2ad23ff195b940f7bd6251b6c 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 c581985b2969488b0222c1310c217bf651a23be3..e778cb8d881dbacd4f87f260bf647559f14d162e 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 faebc855a203882d26b03cd2d559a758eea19062..17e020e9a7e8a0e28ef7ddd0b582952efe25b477 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 ccd4fdf23a8fcc3d62422e91fbba2a300e147253..acc0aa38012ee68c26c64fe28efaf1cf4e6edc10 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 05eaadc2c112b1bbfae6a4aab359aee634988b73..48257bc13dea03bcf82884dc25575904a53a7531 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 9b7f81d9f6a791e4c94e3ecb9520ec2799df9f00..2880583b9991a0d10d54493d8ea2769327b03abb 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 87f9d187271e6c013a7678c92f1de1183ffd3fc3..12891491abca6d2aab61928042c9669c97c3b769 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 a9101b9a916ad67d82d54c20f7b0a31159b2c9fb..aeee054da0640ed35791ad66e8ae2eac97270559 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 4201758d1e9b5ed812254200976a18fa160f7aef..198e4333fae09bef3e2b8c5532ac7b5d173814d0 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 aa28a5db42893ba22ecf146f7cbd54304eaf69ed..2739ea9d12d3e63b7993f40a06098f3a8df7ec00 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 14ab95a7d6e76c9a011a108aee9c053bf31ec6bd..ba49dbbc2e64d67a432d7b0dd60e75b3a5ebdb2e 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 c2fdf5716c82eda03afbbf60684838c04c20922f..4a9ba8b0155a69320861235530c5b19752be16eb 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 e3b9ab4b9876c0ed6d989701d275fe405323796e..3ac7264acb5d01f2ffb064bed45d3e596dea638a 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 e22c32dcc8505558803600df6486e0b04d36325e..10b3c7f9ef6bc41b6d4a95619536aa4f1ca171e9 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 a0bd2e36432327407b75317d8da4ae303165ec08..719a20068e71121375ba7067cd9ec00019e70d54 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 2999115904a8a0fafd1d39290046054b1a7d9974..ff84fadd7c1eabbe9488ca81eaad2f3e807588cb 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 9efae98b2d13527c103d7fdbae53215c0debb24c..31eef0c3537cf82a76a10aec0963d2c07e626cfa 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 f262fda9296e7c334de0fb27f4272ba8a8551425..eb4e3a3bd4dbcf036bb8d71b9a242382cdecce8c 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 3d24e00be941801f8c96e39d92c2b367daad3060..e6b437bb4082f7e0078846a99a6f360836bc07a4 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 7f334b31edb8c3af01d01a440e73fd189a7768e7..66317732ec8b33ea5f485e1901f613a44d009284 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 1bf91e59bee2d16892aadc30b770f8bfc6bcb405..d013e647002bcfc0c79f2a2142f3a4bb06f71722 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 07d64ed4063dfbc61704782decfbaa996050a34c..33b8dabfa4a93c220c9d35cd4323d7892d2cd915 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 3881a4684e34a894684e3a573e1177dfd06ad99e..a3eaf4c58149924f774d0a472d70e7d73720163e 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 60ffdff4c0d0680c09577e4e2ecd3e384000968b..d16151df2512d2a03d6cd10c21e4d73c8b787fc1 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 0f06f369ad957323f9515d44b12b527cf6418ee5..daf7276218dec0fe33731bf5478323b02c5a2942 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 ffccaf8c3e4c5ede2e765edf539de3d208ba529f..873b027a927df96332d2ace9e0da897f3dbcdc92 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 5dad03d99903848bef3ef975d02137dfc8dc2935..7e065cd4075ee232e156a98927205aed2a40d760 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 454f2815dd4483e4353489f4032da6ed79ac716a..e3fad8ff4ef5044fed24c1e081487e7a65343658 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 63c52c0a9f53515b85b97783d2f907fb8ace8d3f..336fb33454bcbc47c6590444eed85c28cab5524b 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 b8976e33172e7fe4452ac2f23f0aefd99313ec42..4981efe91143bd88aee3e96a690ce2f75b2d8b6f 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 f1ae9ee7c7794e5d581cf0fb2200a8a5c38a9688..47aa4c0e4f40b760f21d3361c1902976ebfab84f 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 ec414f2e556f5f6b4d5648a0c679dd13ef0ba442..7f4de66405bf7b9d6a06398bc419774eb7fa128f 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 889bccf55d7ff84471fc7710410fc8e690edf52f..ee606707a34b4491b5801a41f9ecf2708375ac1e 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 3eb06dfec8b8f0b81057cd7465c84834e1eb20bf..835782185b92e999bfdbc3698c0bc70fe3dbcefe 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