Missing callbacks argument to fit method

[hvgazula]

What are you trying to accomplish?

• Adding/passing callbacks to the fit method (for Segmentation)
• Noticed no callbacks input argument in the method definition

What have you tried? (TODO)

• [x] Pass 1 or 2 callbacks
• [x] submit task (waiting for GPU)
• [x] Update code accordingly to enable callbacks and rerun

[hvgazula]

TypeError: Segmentation.fit() got an unexpected keyword argument 'callbacks'

[hvgazula]

• [ ] test on a100 (but otherwise confident about the resolution)
• [ ] add example in guide