Passing checkpoint related arguments from outside Segmentation

[hvgazula]

nobrainer/nobrainer/processing/base.py

Line 28 in 976691d

self.checkpoint_tracker = CheckpointTracker(self, checkpoint_filepath)

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