coqui-ai
/
TTS
mirror of https://github.com/coqui-ai/TTS.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

bug fix

pull/10/head
Eren Golge 2018-05-02 04:59:51 -07:00
parent 855dd365b5
commit 59f8d246c9
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
train.py
View File

@ -95,7 +95,7 @@ def train(model, criterion, criterion_st, data_loader, optimizer, epoch):
optimizer.zero_grad() optimizer.zero_grad()
stop_target = stop_target.view(c.batch_size, stop_target.size(1) // c.r, -1) stop_target = stop_target.view(c.batch_size, stop_target.size(1) // c.r, -1)
stop_target = (stop_target.sum(2) > 0.0).long() stop_target = (stop_target.sum(2) > 0.0).float()
# dispatch data to GPU # dispatch data to GPU
if use_cuda: if use_cuda: