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

stop targets

This commit is contained in:
Eren Golge 2018-04-30 05:07:41 -07:00
parent 67dfb2f01d
commit a3b1db98e9
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
train.py
View File

@ -81,6 +81,7 @@ def train(model, criterion, data_loader, optimizer, epoch):
linear_spec = data[2] linear_spec = data[2]
mel_spec = data[3] mel_spec = data[3]
mel_lengths = data[4] mel_lengths = data[4]
stop_target = data[5]
current_step = num_iter + args.restore_step + \ current_step = num_iter + args.restore_step + \
epoch * len(data_loader) + 1 epoch * len(data_loader) + 1
@ -225,6 +226,7 @@ def evaluate(model, criterion, data_loader, current_step):
linear_spec = data[2] linear_spec = data[2]
mel_spec = data[3] mel_spec = data[3]
mel_lengths = data[4] mel_lengths = data[4]
stop_target = data[5]
# dispatch data to GPU # dispatch data to GPU
if use_cuda: if use_cuda: