mirror of https://github.com/coqui-ai/TTS.git
bug fix
This commit is contained in:
parent
cbcdec83da
commit
c5540d80cc
2
train.py
2
train.py
|
@ -409,7 +409,7 @@ def evaluate(model, criterion, ap, global_step, epoch):
|
||||||
loss_dict['ga_loss'].item(),
|
loss_dict['ga_loss'].item(),
|
||||||
keep_avg['avg_ga_loss'],
|
keep_avg['avg_ga_loss'],
|
||||||
align_score, keep_avg['avg_align_score']),
|
align_score, keep_avg['avg_align_score']),
|
||||||
flush=Tr ue)
|
flush=True)
|
||||||
|
|
||||||
if args.rank == 0:
|
if args.rank == 0:
|
||||||
# Diagnostic visualizations
|
# Diagnostic visualizations
|
||||||
|
|
Loading…
Reference in New Issue