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

Merge branch 'master' of github.com:mozilla/TTS

This commit is contained in:
Eren Golge 2019-01-06 18:13:54 +01:00
parent c8d7a6a84e 2634abbbd5
commit c7ee056f21
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
train.py
View File

@ -106,7 +106,7 @@ def train(model, criterion, criterion_st, optimizer, optimizer_st,
mel_lengths = mel_lengths.cuda(non_blocking=True) mel_lengths = mel_lengths.cuda(non_blocking=True)
linear_input = linear_input.cuda(non_blocking=True) linear_input = linear_input.cuda(non_blocking=True)
stop_targets = stop_targets.cuda(non_blocking=True) stop_targets = stop_targets.cuda(non_blocking=True)
# compute mask for padding # compute mask for padding
mask = sequence_mask(text_lengths) mask = sequence_mask(text_lengths)
@ -385,7 +385,7 @@ def main(args):
optimizer.load_state_dict(checkpoint['optimizer']) optimizer.load_state_dict(checkpoint['optimizer'])
print( print(
" > Model restored from step %d" % checkpoint['step'], flush=True) " > Model restored from step %d" % checkpoint['step'], flush=True)
start_epoch = checkpoint['epoch'] start_epoch = checkpoint['epoch']
best_loss = checkpoint['linear_loss'] best_loss = checkpoint['linear_loss']
args.restore_step = checkpoint['step'] args.restore_step = checkpoint['step']
else: else: