RuntimeError: Error(s) in loading state_dict for DataParallel: Unexpected key(s) in state_dict: "module.albert.embeddings.position_ids"

[mondal-anindita]

Traceback (most recent call last): File "", line 1, in File "/home/anindita/Documents/Speech/indic-punct/src/punctuate/punctuate_text.py", line 32, in init self.tokenizer, self.model, self.train_encoder, self.punctuation_dict = self.load_model_parameters() File "/home/anindita/Documents/Speech/indic-punct/src/punctuate/punctuate_text.py", line 114, in load_model_parameters model.load_state_dict(checkpoint['state_dict']) File "/home/anindita/anaconda3/envs/ip10/lib/python3.10/site-packages/torch/nn/modules/module.py", line 2215, in load_state_dict raise RuntimeError('Error(s) in loading state_dict for {}:\n\t{}'.format( RuntimeError: Error(s) in loading state_dict for DataParallel: Unexpected key(s) in state_dict: "module.albert.embeddings.position_ids"