regisss HF staff commited on
Commit
7a0b241
1 Parent(s): a05195e

Update gaudi_config.json

Browse files
Files changed (1) hide show
  1. gaudi_config.json +1 -0
gaudi_config.json CHANGED
@@ -22,6 +22,7 @@
22
  "conv2d"
23
  ],
24
  "hmp_fp32_ops": [
 
25
  "nll_loss",
26
  "log_softmax",
27
  "embedding"
 
22
  "conv2d"
23
  ],
24
  "hmp_fp32_ops": [
25
+ "cross_entropy",
26
  "nll_loss",
27
  "log_softmax",
28
  "embedding"