In Examples_pytorch.ipynb file unable to find XRAI mask for other classes by replacing call_model_args

[arunsubk]

Hi,

Firstly thanks for this great repo.

In the Examples_pytorch.ipynb file, If i change call_model_args = {class_idx_str: prediction_class} to call_model_args = {class_idx_str: <ANY_OTHER_CLASS>} XRAI mask still remains that of Doberman (i.e. class 236). What do I do to see the visualization of any other class in the Doberman image? I had for instance set ANY_OTHER_CLASS to 562 (fountain) and was hoping to see difference gradient image of the class. But I still see the Doberman.

Could you please help? Thanks,