Fix aux unit tests

This commit is contained in:
Edresson Casanova 2023-12-06 17:47:33 -03:00 committed by Enno Hermann
parent c99e885cc8
commit fc099218df
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,