updates format

cross_entropy_loss.py

@@ -63,8 +63,8 @@ class cross_entropy_loss(evaluate.Metric):
     def _compute(self, prediction_scores, references):
         """Returns the scores"""
 
-        loss = F.cross_entropy(input=torch.from_numpy(prediction_scores).flatten(start_dim=0, end_dim=1),
-                            target=torch.from_numpy(references).flatten(),
+        loss = F.cross_entropy(input=torch.from_numpy(prediction_scores),
+                            target=torch.from_numpy(references),
                             ignore_index=-100).item()
         return {
             "cross_entropy_loss": loss