Habana
regisss HF staff commited on
Commit
1b5a95e
1 Parent(s): 95e6ca0

Create gaudi_config.json

Browse files
Files changed (1) hide show
  1. gaudi_config.json +27 -0
gaudi_config.json ADDED
@@ -0,0 +1,27 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "use_habana_mixed_precision": false,
3
+ "hmp_opt_level": "O1",
4
+ "hmp_is_verbose": false,
5
+ "use_fused_adam": true,
6
+ "use_fused_clip_norm": true,
7
+ "hmp_bf16_ops": [
8
+ "add",
9
+ "addmm",
10
+ "bmm",
11
+ "div",
12
+ "dropout",
13
+ "gelu",
14
+ "iadd",
15
+ "layer_norm",
16
+ "matmul",
17
+ "mm",
18
+ "rsub",
19
+ "softmax"
20
+ ],
21
+ "hmp_fp32_ops": [
22
+ "embedding",
23
+ "linear",
24
+ "nll_loss",
25
+ "log_softmax"
26
+ ]
27
+ }