I am training my own code, and I want to implement one function:
After training n(=100) iterations, I want to see feature maps.
Now I have done with register_forward_hook to see the feature maps, But I want to see them after n iterations. Any idea about how to implement this?
Thanks.