The problem is I am not able to find a way to get grad of conv1 and conv2 from Sequential Block1 and similarly I wanted to track all way along.  Therefore I don’t know what to write in your_func
by mod[0] (mod being a sequential block) will retrieve nn.Conv2d(3,16,3) (in reference to my model var Block1)
Therefore it will be a great help if could help me in an example using my model
Thanks in advance