Habana
regisss HF Staff commited on
Commit
84e72f9
·
1 Parent(s): f7f9092

Back to HMP

Browse files
Files changed (1) hide show
  1. gaudi_config.json +4 -3
gaudi_config.json CHANGED
@@ -1,8 +1,9 @@
1
  {
 
 
2
  "use_fused_adam": true,
3
  "use_fused_clip_norm": true,
4
- "use_torch_autocast": true,
5
- "autocast_bf16_ops": [
6
  "add",
7
  "addmm",
8
  "bmm",
@@ -18,7 +19,7 @@
18
  "softmax",
19
  "truediv"
20
  ],
21
- "autocast_fp32_ops": [
22
  "embedding",
23
  "nll_loss",
24
  "log_softmax",
 
1
  {
2
+ "use_habana_mixed_precision": true,
3
+ "hmp_is_verbose": false,
4
  "use_fused_adam": true,
5
  "use_fused_clip_norm": true,
6
+ "hmp_bf16_ops": [
 
7
  "add",
8
  "addmm",
9
  "bmm",
 
19
  "softmax",
20
  "truediv"
21
  ],
22
+ "hmp_fp32_ops": [
23
  "embedding",
24
  "nll_loss",
25
  "log_softmax",