Tag: pre-trained-model

Accessing a specific layer in a pretrained model in PyTorch

deep-learning pre-trained-model python pytorch

I want to extract the features from certain blocks of the TimeSformer model and also want to remove the last two layers. The print of the model is as follows: ) ) Specifically, I want to extract the outputs of the 4th, 8th and 11th blocks of the model and removing the lats two layers. How can I do this.