error when loading a saved fine tunned model #1507
RafieeAshkan
started this conversation in
General
Replies: 0 comments
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
-
HI,
Thank you for your excellent package. I have fine tuned a roberta model for text classification and saved the fine tuned model. I would like to now load the saved best model and use it for inference. However, I get runtime error as follows:
Here is how I am loading the saved model:
Any help is highly appreciated
Beta Was this translation helpful? Give feedback.
All reactions