regisss HF staff commited on
Commit
dafa7e3
1 Parent(s): a5951b6

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
- "embedding",
24
- "nll_loss",
25
- "log_softmax",
26
- "cross_entropy"
27
- ]
28
- }
 
1
  {
2
+ "use_torch_autocast": true,
 
3
  "use_fused_adam": true,
4
+ "use_fused_clip_norm": true
5
+ }