Grad-CAM from efficientnet-b0 error : forward() takes 1 positional argument but 2 were given

Question

I want to implement grad-cam from efficientnet-b0 model (https://github.com/lukemelas/EfficientNet-PyTorch), but I got an error "forward() takes 1 positional argument but 2