regisss HF staff commited on
Commit
145f34d
1 Parent(s): c7e2e9b

Update gaudi_config.json

Browse files
Files changed (1) hide show
  1. gaudi_config.json +1 -24
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
- "embedding",
24
- "nll_loss",
25
- "log_softmax",
26
- "cross_entropy"
27
- ]
28
  }
 
1
  {
 
 
2
  "use_fused_adam": true,
3
  "use_fused_clip_norm": true,
4
+ "use_torch_autocast": true
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
5
  }