From d11a53e0ea4cbe70f7c5581928e4862080c16320 Mon Sep 17 00:00:00 2001
From: shahules786 <shahules786@gmail.com>
Date: Fri, 14 Oct 2022 12:46:30 +0530
Subject: [PATCH] log test duration

---
 enhancer/models/model.py | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/enhancer/models/model.py b/enhancer/models/model.py
index b283a45..aa0199b 100644
--- 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):