print eval loss

SeanLee97 · Sep 30, 2024 · fccb9d8 · fccb9d8
1 parent 959bf1b
commit fccb9d8
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/angle_emb/angle.py b/angle_emb/angle.py
@@ -862,7 +862,9 @@ def compute_loss(self, model, inputs, return_outputs=False):
     def prediction_step(self, model, inputs, *args, **kwargs):
         kwargs.pop('prediction_loss_only', None)
         inputs.pop('labels', None)
-        return super().prediction_step(model, inputs, prediction_loss_only=True, **kwargs)
+        ret = super().prediction_step(model, inputs, prediction_loss_only=True, **kwargs)
+        print(f'eval loss: {ret[0]}')
+        return ret
 
 
 class AngleESETrainer(AngleTrainer):