Habana
/

bert-large-uncased-whole-word-masking

Model card Files Files and versions Community

regisss HF staff commited on Aug 25, 2022

Commit

a1c79f4

•

1 Parent(s): 274d09d

Update gaudi_config.json

Files changed (1) hide show

gaudi_config.json +1 -0

gaudi_config.json CHANGED Viewed

@@ -21,6 +21,7 @@
     "truediv"
   ],
   "hmp_fp32_ops": [
     "embedding",
     "nll_loss",
     "log_softmax"

     "truediv"
   ],
   "hmp_fp32_ops": [
+    "cross_entropy",
     "embedding",
     "nll_loss",
     "log_softmax"