search

kevinjohncutler / omnipose

Omnipose: a high-precision solution for morphology-independent cell segmentation
https://omnipose.readthedocs.io
Other
85 stars 29 forks source link

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

Open lacan opened 2 months ago

lacan commented 2 months ago

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

kevinjohncutler commented 2 months ago

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