MODEL UNDERSTANDING WITH CAPTUM tutorial: Canceled future for execute_request message before replies were done The Kernel crashed while executing code in the the current cell or a previous cell.

[heavenkiller2018]

Add Link

https://pytorch.org/tutorials/beginner/introyt/captumyt.html#feature-attribution-with-integrated-gradients

Describe the bug

the following error occured when running the captum toturial:

Canceled future for execute_request message before replies were done
The Kernel crashed while executing code in the the current cell or a previous cell. Please review the code in the cell(s) to identify a possible cause of the failure. Click [here](https://aka.ms/vscodeJupyterKernelCrash) for more info. View Jupyter [log](command:jupyter.viewOutput) for further details

Could anyone help to provide?
Thanks!

Describe your environment

Linux; CUDA 12.0; pytorch 2.0.0.post200

cc @suraj813

[guptaaryan16]

Hey @svekars I would like to take up this issue

[QasimKhan5x]

/assigntome