Skip to content
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

availability of pretrained models #4

Closed
Geethen opened this issue May 2, 2023 · 2 comments
Closed

availability of pretrained models #4

Geethen opened this issue May 2, 2023 · 2 comments

Comments

@Geethen
Copy link

Geethen commented May 2, 2023

I may have missed it, but I cannot seem to see the model checkpoints. Could these be made available please?

@gabrieltseng
Copy link
Collaborator

gabrieltseng commented May 2, 2023

Hi!

The trained model is available here: https://github.com/nasaharvest/presto/blob/main/data/default_model.pt

You can load the pretrained model with the following code:

from presto import Presto

encoder_decoder = Presto.load_pretrained()

where load_pretrained just loads the state dict in default_model.pt.

@Geethen
Copy link
Author

Geethen commented May 3, 2023

Hi, Thanks for the heads up:) I missed this. Great work. Thank you for sharing.

@Geethen Geethen closed this as completed May 3, 2023
@gabrieltseng gabrieltseng mentioned this issue May 14, 2023
2 tasks
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants