Added capability of continue training from previous checkpoints

2023-06-12 19:02:32 +08:00
parent 291d8ddf5e
commit 631f97eff7
1 changed files with 1 additions and 0 deletions
@@ -310,6 +310,7 @@ def get_hparams(init=True):
    hparams = HParams(**config)
    hparams.model_dir = model_dir
    hparams.max_epochs = args.max_epochs
+    hparams.cont = args.cont
    hparams.drop_speaker_embed = args.drop_speaker_embed
    return hparams