GradCAM for efficientnet and mobilenet

[chetasanand]

I'm using efficientnet and mobilenet in my project and wanted to validate the predictions with Grad CAM, is this module only compatible with the below architectures? What changes do I need to use this module with efficientnet and mobilenet?

FasterRCNN: model.backbone
Resnet18 and 50: model.layer4[-1]
VGG and densenet161: model.features[-1]
mnasnet1_0: model.layers[-1]
ViT: model.blocks[-1].norm1
SwinT: model.layers[-1].blocks[-1].norm1