From b7fc5df3864e86dd0fdf35f68531f9bf024aa183 Mon Sep 17 00:00:00 2001
From: Alexander Engelsberger <aengelsb@hs-mittweida.de>
Date: Thu, 20 May 2021 13:47:20 +0200
Subject: [PATCH] Log test loss.

---
 prototorch/models/glvq.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/prototorch/models/glvq.py b/prototorch/models/glvq.py
index 65b291f..e0ab42d 100644
--- a/prototorch/models/glvq.py
+++ b/prototorch/models/glvq.py
@@ -95,7 +95,7 @@ class GLVQ(AbstractPrototypeModel):
     def test_step(self, batch, batch_idx):
         # `model.eval()` and `torch.no_grad()` handled by pl
         out, test_loss = self.shared_step(batch, batch_idx)
-        return test_loss
+        self.log_dict({'test_loss': test_loss})
 
     # def predict_step(self, batch, batch_idx, dataloader_idx=None):
     #     pass