I’ve a custom model which I’ve saved as model_dict (which I saved as a .pth file) after training. Now I want to use it as feature extractor. How can I do this?
network architecture 31
Please see the model architecture from the above link.
Specifically I want to extract the features of the colored layer. can someone point out an example or a tutorial to achieve the same. Thanks!