bug fix

2018-05-02 06:07:20 -07:00 · 2018-05-02 06:07:20 -07:00 · 9924ae4a1d
parent 6b937d6059
commit 9924ae4a1d
1 changed files with 3 additions and 0 deletions
--- a/train.py
+++ b/train.py
@ -239,6 +239,9 @@ def evaluate(model, criterion, criterion_st, data_loader, current_step):
            mel_lengths = data[4]
            stop_target = data[5]

+            stop_target = stop_target.view(c.batch_size, stop_target.size(1) // c.r, -1)
+            stop_target = (stop_target.sum(2) > 0.0).float().unsqueeze(2)
+        
            # dispatch data to GPU
            if use_cuda:
                text_input = text_input.cuda()