I have the same error in the example you gave, that is, I have only changed the class now ，new_num_classes = 100`  classes, and now I want to load the model after fine-tuning, but I am getting an error, I don’t know how to change it? Have you solved it?