add test dataloader

mwalmsley · Jan 15, 2024 · e3124b7 · e3124b7
1 parent 372bdb1
commit e3124b7
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/zoobot/pytorch/datasets/webdatamodule.py b/zoobot/pytorch/datasets/webdatamodule.py
@@ -161,6 +161,9 @@ def train_dataloader(self):
 
     def val_dataloader(self):
         return self.make_loader(self.val_urls, mode="val")
+
+    def test_dataloader(self):
+        return self.make_loader(self.val_urls, mode="test")
 
     def predict_dataloader(self):
         return self.make_loader(self.predict_urls, mode="predict")