it ran. now logging back on

mwalmsley · Jan 15, 2024 · 2b387c6 · 2b387c6
1 parent de7562b
commit 2b387c6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/zoobot/pytorch/estimators/define_model.py b/zoobot/pytorch/estimators/define_model.py
@@ -127,7 +127,7 @@ def on_validation_epoch_end(self) -> None:
 
     def on_test_epoch_end(self) -> None:
         logging.info('start test epoch end')
-        # self.log_all_metrics(subset='test')
+        self.log_all_metrics(subset='test')
         logging.info('end test epoch end')
 
     def calculate_loss_and_update_loss_metrics(self, predictions, labels, step_name):