search

nasaharvest / presto

Lightweight, Pre-trained Transformers for Remote Sensing Timeseries
https://arxiv.org/abs/2304.14065
MIT License
151 stars 26 forks source link

availability of pretrained models #4

Closed Geethen closed 1 year ago

Geethen commented 1 year ago

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

gabrieltseng commented 1 year ago

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 commented 1 year ago

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