Add validation and test logic

2021-05-19 16:30:19 +02:00
parent 7700bb7f8d
commit fdf9443a2c
4 changed files with 65 additions and 24 deletions
--- a/prototorch/models/abstract.py
+++ b/prototorch/models/abstract.py
@@ -52,7 +52,7 @@ class SiamesePrototypeModel(pl.LightningModule):
        backbone.

        """
-        # model.eval()  # ?!
+        self.eval()
        with torch.no_grad():
            protos, plabels = self.proto_layer()
            if map_protos: