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

Fix the restore_epoch

This commit is contained in:
Eren 2018-10-25 13:56:33 +02:00
parent c0778fa16d
commit e00c3f5044
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

@ -274,7 +274,7 @@ def evaluate(model, criterion, criterion_st, data_loader, ap, current_step):
const_spec = linear_output[idx].data.cpu().numpy() const_spec = linear_output[idx].data.cpu().numpy()
gt_spec = linear_input[idx].data.cpu().numpy() gt_spec = linear_input[idx].data.cpu().numpy()
align_img = alignments[idx].data.cpu().numpy() align_img = alignments[idx].data.cpu().numpy()
\
const_spec = plot_spectrogram(const_spec, ap) const_spec = plot_spectrogram(const_spec, ap)
gt_spec = plot_spectrogram(gt_spec, ap) gt_spec = plot_spectrogram(gt_spec, ap)
align_img = plot_alignment(align_img) align_img = plot_alignment(align_img)
@ -422,7 +422,7 @@ def main(args):
best_loss = checkpoint['linear_loss'] best_loss = checkpoint['linear_loss']
args.restore_step = checkpoint['step'] args.restore_step = checkpoint['step']
else: else:
args.restore_step = 0 args.restore_step = -1
print("\n > Starting a new training", flush=True) print("\n > Starting a new training", flush=True)
if use_cuda: if use_cuda:
model = model.cuda() model = model.cuda()