From 1e017f866abf5d1249d19ec77a3a91a7b7ff227d Mon Sep 17 00:00:00 2001
From: erogol <erogol@hotmail.com>
Date: Wed, 3 Jun 2020 12:15:48 +0200
Subject: [PATCH] reset loss weights

---
 vocoder/configs/melgan_config.json | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/vocoder/configs/melgan_config.json b/vocoder/configs/melgan_config.json
index 4aeba82e..bdfdf88a 100644
--- a/vocoder/configs/melgan_config.json
+++ b/vocoder/configs/melgan_config.json
@@ -54,11 +54,11 @@
     "use_hinge_gan_loss": false,
     "use_feat_match_loss": false,  // use only with melgan discriminators
 
-    "stft_loss_weight": 1,
-    "subband_stft_loss_weight": 1,
+    "stft_loss_weight": 0.5,
+    "subband_stft_loss_weight": 0.5,
     "mse_gan_loss_weight": 2.5,
     "hinge_gan_loss_weight": 2.5,
-    "feat_match_loss_weight": 10.0,
+    "feat_match_loss_weight": 25,
 
     "stft_loss_params": {
         "n_ffts": [1024, 2048, 512],