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)