-
Notifications
You must be signed in to change notification settings - Fork 293
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Is there anyway that I can save the model for later use #17
Comments
You can use: Then load it to do inference: For more information: https://pytorch.org/tutorials/beginner/saving_loading_models.html |
Have you tried this? Not work for me. Because the torch saving seems to use the pickle too. |
Can you show me your code? |
@binhnguyen7tiki Please post the full error message. @binhna I don't think my implementation includes BTW, it seems the two accounts are the same person? Correct me if I'm wrong. |
Do you use the original KAN (https://github.com/KindXiaoming/pykan) for training your model? Then, ask them there better. |
Seemingly unrelated to this repo. Close for now and if I'm wrong I'll reopen it. |
@Blealtan @hoangthangta sorry for the late reply, and yes, I checked the code again and it was the original KAN indeed. I switched to efficient-kan now and everything seems to be fine. |
Hi, thank you for your work.
As for the title, I would like some ways to save the model for inference. I have tried pickle dump but it does not work.
Thanks
The text was updated successfully, but these errors were encountered: