Log test accuracy.

2021-05-20 14:03:31 +02:00 · 2021-05-20 14:03:31 +02:00 · 0204f5eab6
commit 0204f5eab6
parent b7fc5df386
1 changed files with 2 additions and 0 deletions
--- a/prototorch/models/glvq.py
+++ b/prototorch/models/glvq.py
@ -95,6 +95,8 @@ 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)
+
+        self.log_acc(out, batch[-1], tag="test_acc")
        self.log_dict({'test_loss': test_loss})

    # def predict_step(self, batch, batch_idx, dataloader_idx=None):