From 1daba4371dc0b29e907e43f88ad372b87d9f510b Mon Sep 17 00:00:00 2001 From: erogol Date: Sat, 6 Jun 2020 13:00:46 +0200 Subject: [PATCH] bug fix vocoder model restore --- vocoder/configs/multiband-melgan_and_rwd_config.json | 4 ++-- vocoder/train.py | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/vocoder/configs/multiband-melgan_and_rwd_config.json b/vocoder/configs/multiband-melgan_and_rwd_config.json index c783877d..736f3459 100644 --- a/vocoder/configs/multiband-melgan_and_rwd_config.json +++ b/vocoder/configs/multiband-melgan_and_rwd_config.json @@ -56,8 +56,8 @@ "stft_loss_weight": 0.5, "subband_stft_loss_weight": 0.5, - "mse_gan_loss_weight": 2.5, - "hinge_gan_loss_weight": 2.5, + "mse_G_loss_weight": 2.5, + "hinge_G_loss_weight": 2.5, "feat_match_loss_weight": 25.0, "stft_loss_params": { diff --git a/vocoder/train.py b/vocoder/train.py index 2c3a2e0a..05805c68 100644 --- a/vocoder/train.py +++ b/vocoder/train.py @@ -421,7 +421,7 @@ def main(args): # pylint: disable=redefined-outer-name optimizer_gen.load_state_dict(checkpoint['optimizer']) model_disc.load_state_dict(checkpoint['model_disc']) optimizer_disc.load_state_dict(checkpoint['optimizer_disc']) - except KeyError: + except RuntimeError: print(" > Partial model initialization.") model_dict = model_gen.state_dict() model_dict = set_init_dict(model_dict, checkpoint['model'], c)