How can I get the gradient of each weight matrix after passing an image

Question

I\'m using a pre-trained resnet101 model from Pytorch:

import torch
import matplotlib.pyplot as plt
import numpy as np
import torch.nn as nn
import cv2 as cv