mirror of https://github.com/coqui-ai/TTS.git
against blocking stduot
This commit is contained in:
parent
c562fac7f4
commit
83d7756856
1
train.py
1
train.py
|
@ -258,6 +258,7 @@ def evaluate(model, criterion, data_loader, current_step):
|
||||||
('linear_loss',
|
('linear_loss',
|
||||||
linear_loss.item()),
|
linear_loss.item()),
|
||||||
('mel_loss', mel_loss.item())])
|
('mel_loss', mel_loss.item())])
|
||||||
|
sys.stdout.flush()
|
||||||
|
|
||||||
avg_linear_loss += linear_loss.item()
|
avg_linear_loss += linear_loss.item()
|
||||||
avg_mel_loss += mel_loss.item()
|
avg_mel_loss += mel_loss.item()
|
||||||
|
|
Loading…
Reference in New Issue