How to save inner nn.Sequential layer's output

[ptrblck]

You can use forward hooks to store intermediate activations as shown in this example.

PS: you can post code snippets by wrapping them into three backticks ```, which makes debugging easier.