mirror of https://github.com/coqui-ai/TTS.git
revert debug changes
This commit is contained in:
parent
b94f71c56a
commit
ba08e98bd9
7
train.py
7
train.py
|
@ -465,10 +465,9 @@ def main(args):
|
||||||
best_loss = float('inf')
|
best_loss = float('inf')
|
||||||
|
|
||||||
for epoch in range(0, c.epochs):
|
for epoch in range(0, c.epochs):
|
||||||
# train_loss, current_step = train(model, criterion, criterion_st,
|
train_loss, current_step = train(model, criterion, criterion_st,
|
||||||
# optimizer, optimizer_st, scheduler,
|
optimizer, optimizer_st, scheduler,
|
||||||
# ap, epoch)
|
ap, epoch)
|
||||||
train_loss, current_step = 0, 0
|
|
||||||
val_loss = evaluate(model, criterion, criterion_st, ap, current_step, epoch)
|
val_loss = evaluate(model, criterion, criterion_st, ap, current_step, epoch)
|
||||||
print(
|
print(
|
||||||
" | > Training Loss: {:.5f} Validation Loss: {:.5f}".format(
|
" | > Training Loss: {:.5f} Validation Loss: {:.5f}".format(
|
||||||
|
|
Loading…
Reference in New Issue