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

bug fix

This commit is contained in:
Eren Golge 2018-04-25 08:05:31 -07:00
parent da039707db
commit b2bb98dd6f
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
train.py
View File

@ -78,8 +78,8 @@ def train(model, criterion, data_loader, optimizer, epoch):
# setup input data # setup input data
text_input = data[0] text_input = data[0]
text_lengths = data[1] text_lengths = data[1]
linear_input = data[2] linear_spec = data[2]
mel_input = data[3] mel_spec = data[3]
mel_lengths = data[4] mel_lengths = data[4]
current_step = num_iter + args.restore_step + \ current_step = num_iter + args.restore_step + \
@ -227,8 +227,8 @@ def evaluate(model, criterion, data_loader, current_step):
# setup input data # setup input data
text_input = data[0] text_input = data[0]
text_lengths = data[1] text_lengths = data[1]
linear_input = data[2] linear_spec = data[2]
mel_input = data[3] mel_spec = data[3]
mel_lengths = data[4] mel_lengths = data[4]
# dispatch data to GPU # dispatch data to GPU