omnipose icon indicating copy to clipboard operation
omnipose copied to clipboard

Log the validation loss when using CLI while training an omnipose model

Open lacan opened this issue 1 year ago • 1 comments

Looking through the log and the documentation at https://omnipose.readthedocs.io/training.html

I do not see how I can access the validation loss when training a new omnipose model. It would be necessary to see the performance of my model by looking at how both the training loss and validation loss change over time per epoch.

Could this please be added to the verbose outpus of the log when training an omnipose model using the CLI?

Otherwise, how do I access it via Python?

This is somewhat linked to #16

lacan avatar Apr 08 '24 15:04 lacan

I agree that the validation loss, if computed, should be added to the logs.

kevinjohncutler avatar Apr 11 '24 08:04 kevinjohncutler