How to compare a custom CNN model (and not only resnet)

[tushar-semwal]

Is your feature request related to a problem? Please describe.
I would like to compare two simple vanilla CNN models trained on different datasets. How much changes do I need to do in the current code.

Describe the solution you'd like
Comparing two simple vanilla CNN models trained on different datasets (e.g., MNIST and SVHN)

Describe alternatives you've considered
A clear and concise description of any alternative solutions or features you've considered.

Additional context
Add any other context or screenshots about the feature request here.

[shuaitang]

Hi Tushar,

Thanks for your question!

The only function you need to change is forward_with_gradient_hooks. One can replace this function with the forward function in a customised neural network, and add a gradient hook to the feature map which one would like to compute the gradient map with respect to.

Let us know if you need more information.

[tushar-semwal]

Hi Shuai,

I have sent you a google colab on my changes in the code. I am not sure if the results are correct. Please have a look. Thanks.