regisss HF staff commited on
Commit
8f1f2c6
1 Parent(s): b8b9073

Update gaudi_config.json

Browse files
Files changed (1) hide show
  1. gaudi_config.json +3 -26
gaudi_config.json CHANGED
@@ -1,28 +1,5 @@
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",
10
- "div",
11
- "dropout",
12
- "gelu",
13
- "iadd",
14
- "linear",
15
- "layer_norm",
16
- "matmul",
17
- "mm",
18
- "rsub",
19
- "softmax",
20
- "truediv"
21
- ],
22
- "hmp_fp32_ops": [
23
- "cross_entropy",
24
- "embedding",
25
- "nll_loss",
26
- "log_softmax"
27
- ]
28
  }
 
1
  {
2
+ "use_fused_adam": true,
3
+ "use_fused_clip_norm": true,
4
+ "use_torch_autocast": true
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
5
  }