Bug fixe on unit tests

This commit is contained in:
Edresson Casanova 2023-12-26 09:30:27 -03:00
parent e741cab78e
commit 62b7547033
1 changed files with 1 additions and 1 deletions

View File

@ -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,