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

fix duration estimation

This commit is contained in:
shahules786 2022-10-16 11:13:44 +05:30
parent 0910d9ac84
commit d99fd0eb61
1 changed files with 9 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
enhancer/models/model.py
View File

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