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,