Skip to content
Snippets Groups Projects

Fix error in trainer function

Merged Driss KHALIL requested to merge Fix_checkpoint_process_error into master
2 unresolved threads
@@ -326,6 +326,10 @@ def checkpointer_process(
max_epoch : int
end_potch
Returns
-------
lowest_validation_loss : float
"""
@@ -342,6 +346,8 @@ def checkpointer_process(
if epoch >= max_epoch:
checkpointer.save("model_final_epoch", **arguments)
return lowest_validation_loss
Please register or sign in to reply
def write_log_info(
epoch,
@@ -554,7 +560,7 @@ def run(
samples, model, valid_losses, device, criterion
)
checkpointer_process(
lowest_validation_loss = checkpointer_process(
Please register or sign in to reply
checkpointer,
checkpoint_period,
valid_losses,
Loading