Get output of intermediate layer of pre-trained model

[ptrblck]

Forward hooks are the right approach as described here.