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:43:47 +05:30
parent 315d646347 4f25690e2a
commit 807f4b93ea
1 changed files with 2 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
enhancer/models/model.py
View File

@ -115,16 +115,12 @@ class Model(pl.LightningModule):
self.dataset.setup(stage)
print(
"Total train duration",
self.dataset.train_dataloader().__len__()
* self.dataset.duration
/ 60,
self.dataset.train_dataloader().dataset.__len__() / 60,
"minutes",
)
print(
"Total validation duration",
self.dataset.val_dataloader().__len__()
* self.dataset.duration
/ 60,
self.dataset.val_dataloader().dataset.__len__() / 60,
"minutes",
)
self.dataset.model = self