[FEATURE] Integrate Captum Interpretability as Callback.
🚀 Feature

Motivation

Here is a Captum example: https://captum.ai/tutorials/Resnet_TorchVision_Interpret
Example: callback = CallbackInterpretability(algorithm="integrated_gradients", backend="captum", frequency=5, grid_size=5). And it would create this kind of images every 5 epochs for grid_size * grid_size images.

Pitch

Alternatives

Additional context