From 6dba9a342469574ecbb3660467ebcb701360fad5 Mon Sep 17 00:00:00 2001 From: muskonu <3239089828@qq.com> Date: Sat, 25 Jan 2025 17:40:45 +0800 Subject: [PATCH] fix: the json format of the training imagenet configuration file --- training/imagenet/config/ds_config.json | 2 +- training/imagenet/config/ds_fp16_config.json | 2 +- training/imagenet/config/ds_fp16_z1_config.json | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/training/imagenet/config/ds_config.json b/training/imagenet/config/ds_config.json index d786b90e6..bab0ea4c2 100644 --- a/training/imagenet/config/ds_config.json +++ b/training/imagenet/config/ds_config.json @@ -22,7 +22,7 @@ "zero_allow_untested_optimizer": true, "fp16": { "enabled": false - } + }, "gradient_clipping": 0, "prescale_gradients": false, "cuda_visible_devices": 0, diff --git a/training/imagenet/config/ds_fp16_config.json b/training/imagenet/config/ds_fp16_config.json index 60e28fc8b..646a5d650 100644 --- a/training/imagenet/config/ds_fp16_config.json +++ b/training/imagenet/config/ds_fp16_config.json @@ -23,7 +23,7 @@ "fp16": { "enabled": true, "auto_cast": true - } + }, "gradient_clipping": 0, "prescale_gradients": false, "cuda_visible_devices": 0, diff --git a/training/imagenet/config/ds_fp16_z1_config.json b/training/imagenet/config/ds_fp16_z1_config.json index b3c0d1c68..ae34e5d57 100644 --- a/training/imagenet/config/ds_fp16_z1_config.json +++ b/training/imagenet/config/ds_fp16_z1_config.json @@ -23,7 +23,7 @@ "fp16": { "enabled": true, "auto_cast": true - } + }, "gradient_clipping": 0, "prescale_gradients": false, "cuda_visible_devices": 0,