tuna2134/sbv2-v2
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

Update loss

Browse Source
This commit is contained in:
Stardust·减
2023-11-07 12:58:03 +08:00
committed by GitHub
parent c864828afa
commit 8ea241b57a
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
train_ms.py
View File

@@ -375,6 +375,7 @@ def train_and_evaluate(
z_mask, z_mask,
(z, z_p, m_p, logs_p, m_q, logs_q), (z, z_p, m_p, logs_p, m_q, logs_q),
(hidden_x, logw, logw_), (hidden_x, logw, logw_),
loss_commit,
) = net_g( ) = net_g(
x, x,
x_lengths, x_lengths,
@@ -457,7 +458,9 @@ def train_and_evaluate(
loss_fm = feature_loss(fmap_r, fmap_g) loss_fm = feature_loss(fmap_r, fmap_g)
loss_gen, losses_gen = generator_loss(y_d_hat_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: if net_dur_disc is not None:
loss_dur_gen, losses_dur_gen = generator_loss(y_dur_hat_g) loss_dur_gen, losses_dur_gen = generator_loss(y_dur_hat_g)
loss_gen_all += loss_dur_gen loss_gen_all += loss_dur_gen