Solved by ptrblck in post #4 
              
              
                You would just have to load the pre-trained model and change the last linear layer to the new number of classes as described in the Transfer Learning tutorial.