Skip to content

Accessing the best validation loss so far in validation_epoch_ends #6628

Discussion options

You must be logged in to vote

I think if you include a ModelCheckpoint in your trainer, you should be able to retrieve the best model score. But you should set your validation loss as the monitored variable in your ModelCheckpoint callback. From your validation_epoch_end, you should be able to retrieve your best score by reaching the callback with something similar to self.trainer.callback.best_model_score.

Replies: 1 comment 1 reply

Comment options

You must be logged in to vote
1 reply
@jandono
Comment options

Answer selected by carmocca
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment