I’m playing around with the pretrained models.
I know a network like alexnet has 4096 features before classification. Is there a way in pytorch to see what the number of features are for all the different pretrained networks using the same code? For example I’m looking at resnet152 and cant tell from reading the code here how many features it has : https://github.com/pytorch/vision/blob/master/torchvision/models/resnet.py
You could print(your_model).
It will display the detail information of every layer.