Solved by tom in post #2 
              
              
                when you print the model (print(model)) you should see that there is a model.fc layer. You can make your new nn.Linear and assign it to model.fc.