Solved by Nikronic in post #2 
              
              
                Hi, 
Yes, because squeezenet does not use Linear in its definition so there is no need to define the structure like ResNet or Inception. But you can get the structure of model using print(model) and extract layers that you want. 
Based on your example code, I think you looking for dropping final cla…