ResNet18 - access to the output of each BasicBlock

2

Have a look at this post to see how to use forward hooks to get the output of some internal modules.