I'm encountering a dimension mismatch error while loading the weights for the teacher model (tmodel). The issue specifically occurs in the transformer.embeddings.word_embeddings.weight and mlp.layers. I have used the weight file you provided (iu_t_model.pth), but there is such an error, and there is another weight file (iu_weight.pth) that I have not used. May I ask where it is loaded?
I'm encountering a dimension mismatch error while loading the weights for the teacher model (tmodel). The issue specifically occurs in the transformer.embeddings.word_embeddings.weight and mlp.layers. I have used the weight file you provided (iu_t_model.pth), but there is such an error, and there is another weight file (iu_weight.pth) that I have not used. May I ask where it is loaded?