Medix-AI/coqui-tts
Medix-AI
/
coqui-tts
mirror of https://github.com/coqui-ai/TTS.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Fix scheduler order

This commit is contained in:
Eren Gölge 2022-12-05 12:26:15 +01:00 committed by GitHub
parent c50d89fcf7
commit 9321b22203
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
TTS/tts/models/vits.py
View File

@ -1671,8 +1671,8 @@ class Vits(BaseTTS):
Returns: Returns:
List: Schedulers, one for each optimizer. List: Schedulers, one for each optimizer.
""" """
scheduler_G = get_scheduler(self.config.lr_scheduler_gen, self.config.lr_scheduler_gen_params, optimizer[0]) scheduler_D = get_scheduler(self.config.lr_scheduler_disc, self.config.lr_scheduler_disc_params, optimizer[0])
scheduler_D = get_scheduler(self.config.lr_scheduler_disc, self.config.lr_scheduler_disc_params, optimizer[1]) scheduler_G = get_scheduler(self.config.lr_scheduler_gen, self.config.lr_scheduler_gen_params, optimizer[1])
return [scheduler_D, scheduler_G] return [scheduler_D, scheduler_G]
def get_criterion(self): def get_criterion(self):