Getting output of intermediate layers for a test input

[tataganesh]

This answer from the forums provides a simple implementation of a forward hook to obtain the output of a particular model layer.

Forward hook activations for loss computation

@Alexander_Riedel a very interesting use case. I would definitely like to know the reasons why you want the same feature map for two successive conv layers You will have to make some tweaks to the code For L1 Loss, both the outputs need to be the same, so you need to ensure that the number of channels are the same You need to resize the smaller width, height to the larger width, height so that you can pass that to the L1 Loss. You can leverage torch resize for this In your hook functio…