I’m using it for GradCAM. I train multiple models => save them => load all of them in a new script and run forward pass => get gradients and plot gradcam heatmaps.
That’s why I need those parameters. Is there a better way to do this?
Also, how would I go about saving the parameters I use register_hook to extract into register_buffer? And once stored, how I would I go about loading my model?