Visualization bug with non identified keyword by pyplot...
edited The issue arises because of matplotlib update I just encountered this tonight and I fixed it by editing site-packages/captum/attr/_utils/visualization.py and changed line 250 from plt_axis.grid(b=False) to plt_axis.grid(visible=False) ...