skaulintel commited on
Commit
adc2b67
1 Parent(s): 9275f61

Enable torch autocast for albert large.

Browse files

set use_torch_autocast to true, disable use_habana_mixed_precision

Files changed (1) hide show
  1. gaudi_config.json +2 -1
gaudi_config.json CHANGED
@@ -1,5 +1,6 @@
1
  {
2
- "use_habana_mixed_precision": true,
 
3
  "hmp_is_verbose": false,
4
  "use_fused_adam": true,
5
  "use_fused_clip_norm": true,
1
  {
2
+ "use_habana_mixed_precision": false,
3
+ "use_torch_autocast": true,
4
  "hmp_is_verbose": false,
5
  "use_fused_adam": true,
6
  "use_fused_clip_norm": true,