From 8ea241b57ad9d5479418ced7682acc01e6ebde38 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Stardust=C2=B7=E5=87=8F?= Date: Tue, 7 Nov 2023 12:58:03 +0800 Subject: [PATCH] Update loss --- train_ms.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/train_ms.py b/train_ms.py index 67bd1dd..3c25e32 100644 --- a/train_ms.py +++ b/train_ms.py @@ -375,6 +375,7 @@ def train_and_evaluate( z_mask, (z, z_p, m_p, logs_p, m_q, logs_q), (hidden_x, logw, logw_), + loss_commit, ) = net_g( x, x_lengths, @@ -457,7 +458,9 @@ def train_and_evaluate( loss_fm = feature_loss(fmap_r, fmap_g) loss_gen, losses_gen = generator_loss(y_d_hat_g) - loss_gen_all = loss_gen + loss_fm + loss_mel + loss_dur + loss_kl + loss_gen_all = ( + loss_gen + loss_fm + loss_mel + loss_dur + loss_kl + loss_commit + ) if net_dur_disc is not None: loss_dur_gen, losses_dur_gen = generator_loss(y_dur_hat_g) loss_gen_all += loss_dur_gen