Visualize activations

You could use forward hooks to grab the desired activation maps and visualize them as described here. Alternatively, you could also check other repositories, but I’m unsure how compatible these are to the current PyTorch release (e.g. this one).

I would also generally not recommend tagging specific users in questions, as it could demotivate others to post valid answers.