wassname/TTS
1
0
Fork 0
mirror of https://github.com/wassname/TTS.git synced 2026-09-09 11:16:00 +08:00
Code Issues Packages Projects Releases Wiki Activity

plot alignment score for eval

Browse Source
This commit is contained in:
root
2020-01-03 00:30:01 +01:00
parent 7b6fe29f00
commit 9d666af715
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
train.py
+2 -1
View File
@@ -455,7 +455,8 @@ def evaluate(model, criterion, criterion_st, ap, global_step, epoch):
epoch_stats = {
"loss_postnet": keep_avg['avg_postnet_loss'],
"loss_decoder": keep_avg['avg_decoder_loss'],
"stop_loss": keep_avg['avg_stop_loss']
"stop_loss": keep_avg['avg_stop_loss'],
"alignment_score": keep_avg['avg_align_score']
}
if c.bidirectional_decoder: