update

loss_metric.py

@@ -51,7 +51,7 @@ Examples:
 
 
 @evaluate.utils.file_utils.add_start_docstrings(_DESCRIPTION, _KWARGS_DESCRIPTION)
-class loss_metric(evaluate.Metric):
+class LossMetric(evaluate.Metric):
     """Calculation of the cross-entropy loss function using the huggingface evaluate module."""
 
     def _info(self):
@@ -73,8 +73,9 @@ class loss_metric(evaluate.Metric):
             reference_urls=["http://path.to.reference.url/new_module"]
         )
 
-    def _compute(self, predictions: Tensor, references: Tensor):
+    def _compute(self, predictions, references):
         """Returns the scores"""
+        print(predictions)
         loss_func = nn.CrossEntropyLoss()
         loss = loss_func(predictions, references)