Extracting and using features from a pretrained model

14

An example towards how to use forward_hooks is given in this post