regisss HF staff commited on
Commit
14f454d
1 Parent(s): 3370abb

Add cross_entropy to fp32 ops

Browse files
Files changed (1) hide show
  1. gaudi_config.json +2 -1
gaudi_config.json CHANGED
@@ -24,6 +24,7 @@
24
  "hmp_fp32_ops": [
25
  "log_softmax",
26
  "embedding",
27
- "binary_cross_entropy"
 
28
  ]
29
  }
 
24
  "hmp_fp32_ops": [
25
  "log_softmax",
26
  "embedding",
27
+ "binary_cross_entropy",
28
+ "cross_entropy"
29
  ]
30
  }