Skip to content

Commit

Permalink
fix: the json format of the training imagenet configuration file
Browse files Browse the repository at this point in the history
  • Loading branch information
navanis committed Jan 25, 2025
1 parent a85b5e6 commit 6dba9a3
Show file tree
Hide file tree
Showing 3 changed files with 3 additions and 3 deletions.
2 changes: 1 addition & 1 deletion training/imagenet/config/ds_config.json
Original file line number Diff line number Diff line change
Expand Up @@ -22,7 +22,7 @@
"zero_allow_untested_optimizer": true,
"fp16": {
"enabled": false
}
},
"gradient_clipping": 0,
"prescale_gradients": false,
"cuda_visible_devices": 0,
Expand Down
2 changes: 1 addition & 1 deletion training/imagenet/config/ds_fp16_config.json
Original file line number Diff line number Diff line change
Expand Up @@ -23,7 +23,7 @@
"fp16": {
"enabled": true,
"auto_cast": true
}
},
"gradient_clipping": 0,
"prescale_gradients": false,
"cuda_visible_devices": 0,
Expand Down
2 changes: 1 addition & 1 deletion training/imagenet/config/ds_fp16_z1_config.json
Original file line number Diff line number Diff line change
Expand Up @@ -23,7 +23,7 @@
"fp16": {
"enabled": true,
"auto_cast": true
}
},
"gradient_clipping": 0,
"prescale_gradients": false,
"cuda_visible_devices": 0,
Expand Down

0 comments on commit 6dba9a3

Please sign in to comment.