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

gan vocoder train fix in case of restoring models wiht no scheduler is defined

This commit is contained in:
Eren Gölge 2021-04-04 03:54:20 +02:00
parent 3835a961ca
commit 6ad4eba678
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/bin/train_vocoder_gan.py
View File

@ -497,12 +497,12 @@ def main(args): # pylint: disable=redefined-outer-name
model_disc.load_state_dict(checkpoint['model_disc']) model_disc.load_state_dict(checkpoint['model_disc'])
print(" > Restoring Discriminator Optimizer...") print(" > Restoring Discriminator Optimizer...")
optimizer_disc.load_state_dict(checkpoint['optimizer_disc']) optimizer_disc.load_state_dict(checkpoint['optimizer_disc'])
if 'scheduler' in checkpoint: if 'scheduler' in checkpoint and scheduler_gen is not None:
print(" > Restoring Generator LR Scheduler...") print(" > Restoring Generator LR Scheduler...")
scheduler_gen.load_state_dict(checkpoint['scheduler']) scheduler_gen.load_state_dict(checkpoint['scheduler'])
# NOTE: Not sure if necessary # NOTE: Not sure if necessary
scheduler_gen.optimizer = optimizer_gen scheduler_gen.optimizer = optimizer_gen
if 'scheduler_disc' in checkpoint: if 'scheduler_disc' in checkpoint and scheduler_disc is not None:
print(" > Restoring Discriminator LR Scheduler...") print(" > Restoring Discriminator LR Scheduler...")
scheduler_disc.load_state_dict(checkpoint['scheduler_disc']) scheduler_disc.load_state_dict(checkpoint['scheduler_disc'])
scheduler_disc.optimizer = optimizer_disc scheduler_disc.optimizer = optimizer_disc