bug fix, average mel spec validation loss

2018-05-15 07:13:46 -07:00 · 2018-05-15 07:13:46 -07:00 · a31e60e928
parent b2e09f9f58
commit a31e60e928
3 changed files with 277 additions and 147 deletions
--- a/layers/tacotron.py
+++ b/layers/tacotron.py
@ -315,7 +315,7 @@ class Decoder(nn.Module):
                if t >= T_decoder:
                    break
            else:
-                if t > 1 and stop_token > 0.5:
+                if t > 1 and stop_token > 0.8:
                    break
                elif t > self.max_decoder_steps:
                    print(" !! Decoder stopped with 'max_decoder_steps'. \
--- a/notebooks/TacotronPlayGround.ipynb
+++ b/notebooks/TacotronPlayGround.ipynb
--- a/train.py
+++ b/train.py
@ -300,6 +300,7 @@ def evaluate(model, criterion, criterion_st, data_loader, current_step):

    # compute average losses
    avg_linear_loss /= (num_iter + 1)
+    avg_mel_loss /= (num_iter + 1)
    avg_stop_loss /= (num_iter + 1)
    avg_total_loss = avg_mel_loss + avg_linear_loss + stop_loss