bug fix, average mel spec validation loss

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'. \

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