mirror of https://github.com/coqui-ai/TTS.git
Fix aux unit tests
This commit is contained in:
parent
c99e885cc8
commit
fc099218df
|
@ -248,7 +248,7 @@ def train(model, optimizer, scheduler, criterion, data_loader, eval_data_loader,
|
||||||
)
|
)
|
||||||
# save the best checkpoint
|
# save the best checkpoint
|
||||||
best_loss = save_best_model(
|
best_loss = save_best_model(
|
||||||
eval_loss,
|
{"train_loss": None, "eval_loss": eval_loss},
|
||||||
best_loss,
|
best_loss,
|
||||||
c,
|
c,
|
||||||
model,
|
model,
|
||||||
|
|
Loading…
Reference in New Issue