Solved by Dwight_Foster in post #4 
              
              
                Ok to get just the convolutional layers for alexnet you could do something like this 
model = torchvision.models.alexnet()
model.features

the model.features line will get only the convolutional layers and you can define that as a new model or just pass the input directly into it.