deepgeek/Matcha-TTS
2
0
Fork 0
mirror of https://github.com/shivammehta25/Matcha-TTS.git synced 2026-02-05 18:29:19 +08:00
Code Issues Packages Projects Releases Wiki Activity

Logging global step while training

Browse Source
This commit is contained in:
Shivam Mehta
2023-12-06 10:39:54 +00:00
parent f39ee6cf3b
commit 0ed9290c31
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
matcha/models/baselightningmodule.py
View File

@@ -81,7 +81,7 @@ class BaseLightningClass(LightningModule, ABC):
"step", "step",
float(self.global_step), float(self.global_step),
on_step=True, on_step=True,
on_epoch=True, prog_bar=True,
logger=True, logger=True,
sync_dist=True, sync_dist=True,
) )