julius/prototorch_models
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Log test accuracy.

Browse Source
This commit is contained in:
Alexander Engelsberger 2021-05-20 14:03:31 +02:00
parent b7fc5df386
commit 0204f5eab6
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
prototorch/models/glvq.py
View File

@ -95,6 +95,8 @@ class GLVQ(AbstractPrototypeModel):
def test_step(self, batch, batch_idx): def test_step(self, batch, batch_idx):
# `model.eval()` and `torch.no_grad()` handled by pl # `model.eval()` and `torch.no_grad()` handled by pl
out, test_loss = self.shared_step(batch, batch_idx) 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}) self.log_dict({'test_loss': test_loss})
# def predict_step(self, batch, batch_idx, dataloader_idx=None): # def predict_step(self, batch, batch_idx, dataloader_idx=None):