From ef6ff4e95c201350306333e6d8c89113ecd3997d Mon Sep 17 00:00:00 2001 From: rishikksh20 Date: Wed, 3 Mar 2021 22:31:39 +0530 Subject: [PATCH] Add Exponential LR scheduler check --- TTS/bin/train_vocoder_gan.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/TTS/bin/train_vocoder_gan.py b/TTS/bin/train_vocoder_gan.py index 9c0764fb..2e488e5c 100644 --- a/TTS/bin/train_vocoder_gan.py +++ b/TTS/bin/train_vocoder_gan.py @@ -506,6 +506,8 @@ def main(args): # pylint: disable=redefined-outer-name print(" > Restoring Discriminator LR Scheduler...") scheduler_disc.load_state_dict(checkpoint['scheduler_disc']) scheduler_disc.optimizer = optimizer_disc + if c.lr_scheduler_disc == "ExponentialLR": + scheduler_disc.last_epoch = checkpoint['epoch'] except RuntimeError: # restore only matching layers. print(" > Partial model initialization...")