Can I use imported ClassifierOutputTarget for my fine-tuned model?

[annasli378]

Hi!

I have ResNet model then fine-tuned on a medical images dataset. My model has 4 outputs.
I imported ClassifierOutputTarget as in:

from pytorch_grad_cam.utils.model_targets import ClassifierOutputTarget

but can I use it for my model like this:

targets = [ClassifierOutputTarget(0)]

Would I get target [0] from orginal ResNet model or from my changed model?

Should I create custom ClassifierOutputTarget for my model?

I know I can pass None as targets and it will get target from outputs but I wish I could check how heatmaps would change depending on given target for my model.

Thank you in advance.