From c8953f4da95bde7972a4d0249d6be05e302fff7e Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Eren=20G=C3=B6lge?= <egolge@coqui.ai>
Date: Wed, 7 Apr 2021 12:36:36 +0200
Subject: [PATCH] l1_spec_loss parameters in melgan test config

---
 .../test_vocoder_multiband_melgan_config.json       | 13 +++++++++++++
 1 file changed, 13 insertions(+)

diff --git a/tests/inputs/test_vocoder_multiband_melgan_config.json b/tests/inputs/test_vocoder_multiband_melgan_config.json
index 92deaee4..c1797888 100644
--- a/tests/inputs/test_vocoder_multiband_melgan_config.json
+++ b/tests/inputs/test_vocoder_multiband_melgan_config.json
@@ -49,6 +49,7 @@
     "use_mse_gan_loss": true,
     "use_hinge_gan_loss": false,
     "use_feat_match_loss": false,  // use only with melgan discriminators
+    "use_l1_spec_loss": true,
 
     // loss weights
     "stft_loss_weight": 0.5,
@@ -56,6 +57,7 @@
     "mse_G_loss_weight": 2.5,
     "hinge_G_loss_weight": 2.5,
     "feat_match_loss_weight": 25,
+    "l1_spec_loss_weight": 2.5,
 
     // multiscale stft loss parameters
     "stft_loss_params": {
@@ -71,6 +73,17 @@
         "win_lengths": [150, 300, 60]
     },
 
+    "l1_spec_loss_params": {
+        "use_mel": true,
+        "sample_rate": 22050,
+        "n_fft": 1024,
+        "hop_length": 256,
+        "win_length": 1024,
+        "n_mels": 80,
+        "mel_fmin": 0.0,
+        "mel_fmax": null
+    },
+
     "target_loss": "avg_G_loss",  // loss value to pick the best model to save after each epoch
 
     // DISCRIMINATOR