Passing checkpoint related arguments from outside Segmentation - Githubissues

neuronets / nobrainer

A framework for developing neural network models for 3D image processing.

Other

158 stars 45 forks source link

Passing checkpoint related arguments from outside Segmentation #333

Open hvgazula opened 6 months ago

hvgazula commented 6 months ago

https://github.com/neuronets/nobrainer/blob/976691d685824fd4bba836498abea4184cffd798/nobrainer/processing/base.py#L28

What would you like changed/added and why?

Similar to model_args, I'd like to see checkpoint_args passed from outside.

What would be the benefit?

For example, if I want to save the best model, I prefer setting save_best_only=True. Currently, there is no option to do so unless manually added.

Does the change make something easier to use?

Enables extended capabilities of CheckpointTracker because it extends ModelCheckPoint