Extract the feature output of the own model

Pytorch beginner, my English is not good, I will try to express, I use my own image data to train a model of my own, how to load the model, and input the data into the trained model, extract the final convolution of the model The output characteristics of the layer?

I suggest pytorch beginners to go through the official tutorials for once. Your question is very common so if you done the tutorial, it will be solved. https://pytorch.org/tutorials/

1 Like