Solved by MariosOreo in post #2 
              
              
                Hello, 
You could have a try on model.load_state_dict with strict=False, and define the last fc layer for a different number of classes in addition. load_state_dict