skaulintel commited on
Commit
3f1f93a
1 Parent(s): 995ae44

Enable torch autocast for albert

Browse files
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,