Feature map from r3d_18

2

To visualize the feature (or activation) maps, you could use forward hooks as given in this example. Let me know, if that works for you.