Sorry, I’m very noobie in this yet, so I didn´t understand what you are suggesting.
model = CNN()

model.load_state_dict(torch.load(path))

To extract the features, what do you suggest to do after load the model?