how to get features from efficiennet?

[mosheliv]

Hi,

I am used to use something like:

        self.model = timm.create_model(model_name, pretrained=pretrained)
        n_features = self.model.fc.in_features
        self.model.fc = nn.Linear(n_features, CFG.target_size)

with resnext but it fails with efficientnet. I read about the new features extraction interface, but couldn't get it to work.
Any help will be greatly appreciated.

[ghnreigns]

I think for efficientnet the last layer is called classifier