From 2b387c65ceb7a12e52f46e5c3056570f291fc34a Mon Sep 17 00:00:00 2001 From: Mike Walmsley Date: Mon, 15 Jan 2024 15:34:42 -0500 Subject: [PATCH] it ran. now logging back on --- zoobot/pytorch/estimators/define_model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/zoobot/pytorch/estimators/define_model.py b/zoobot/pytorch/estimators/define_model.py index 2524a116..1cb6e237 100755 --- 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):