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)

• Pass 1 or 2 callbacks
• submit task (waiting for GPU)
• 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