log test duration

2022-10-14 12:46:30 +05:30 · 2022-10-14 12:46:30 +05:30 · d11a53e0ea
parent 4f25690e2a
commit d11a53e0ea
1 changed files with 5 additions and 0 deletions
--- a/enhancer/models/model.py
+++ b/enhancer/models/model.py
@ -123,6 +123,11 @@ class Model(pl.LightningModule):
                self.dataset.val_dataloader().dataset.__len__() / 60,
                "minutes",
            )
+            print(
+                "Total test duration",
+                self.dataset.test_dataloader().dataset.__len__() / 60,
+                "minutes",
+            )
            self.dataset.model = self

    def train_dataloader(self):