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 eval iter

This commit is contained in:
Eren Gölge 2021-03-24 12:31:21 +01:00
parent a3a840fd78
commit b4c2cf80f2
1 changed files with 14 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
TTS/bin/train_align_tts.py
View File

@ -309,19 +309,25 @@ if __name__ == '__main__':
# forward pass model
with torch.cuda.amp.autocast(enabled=c.mixed_precision):
decoder_output, dur_output, dur_mas_output, alignments, mu, log_sigma, logp_max_path = model.forward(
decoder_output, dur_output, dur_mas_output, alignments, _, _, logp = model.forward(
text_input,
text_lengths,
mel_targets,
mel_lengths,
g=speaker_c)
g=speaker_c,
phase=training_phase)
# compute loss
loss_dict = criterion(logp,
decoder_output,
mel_targets,
mel_lengths,
dur_output,
dur_mas_output,
text_lengths,
global_step,
phase=training_phase)
# compute loss
loss_dict = criterion(mu, log_sigma, logp_max_path,
decoder_output, mel_targets,
mel_lengths, dur_output,
dur_mas_output, text_lengths,
global_step, training_phase)
# step time
step_time = time.time() - start_time