You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
I'm getting the following error when attempting to use the downloaded PR_model and replacing the best_model.pt with PR_model_MSE.pt in model_path in inference.py.
Error(s) in loading state_dict for RobertaForSequenceClassification:
Missing key(s) in state_dict: "roberta.embeddings.position_ids", "roberta.embeddings.word_embeddings.weight",
[...]
Please note that I have made sure of making a separate environment and installing all the requirements from requirements.txt.
I'm getting the following error when attempting to use the downloaded PR_model and replacing the
best_model.pt
withPR_model_MSE.pt
inmodel_path
in inference.py.Please note that I have made sure of making a separate environment and installing all the requirements from
requirements.txt
.Full error:
The text was updated successfully, but these errors were encountered: