Seich
/
mayavoz
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Merge branch 'dev' of https://github.com/shahules786/enhancer into dev-hawk

This commit is contained in:
shahules786 2022-10-14 12:46:48 +05:30
parent 0e58691a2c d11a53e0ea
commit 6e0f69f575
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
enhancer/models/model.py
View File

@ -123,6 +123,11 @@ class Model(pl.LightningModule):
self.dataset.val_dataloader().dataset.__len__() / 60, self.dataset.val_dataloader().dataset.__len__() / 60,
"minutes", "minutes",
) )
print(
"Total test duration",
self.dataset.test_dataloader().dataset.__len__() / 60,
"minutes",
)
self.dataset.model = self self.dataset.model = self
def train_dataloader(self): def train_dataloader(self):