search

gastruc / OmniSat

MIT License
56 stars 2 forks source link

pretrained model #1

Open mzyyyyyyyy opened 2 months ago

mzyyyyyyyy commented 2 months ago

could you please upload pretrained model weights(checkpoint)? i want to evaluate the model first. thank you!😀

NielsRogge commented 1 month ago

Subscribing to this thread, happy to make it available on Hugging Face, and link it to the paper page: https://huggingface.co/papers/2404.08351.

gastruc commented 1 month ago

Hi, I will make them available in next weeks. There will be one pretrained model by dataset. As we say in the paper, we pretrain and fine tune on the same dataset and the use of pretrained weights on an other dataset is not trivial.

NielsRogge commented 1 month ago

Cool! Leaving a guide on how to push it to the 🤗 hub for better discoverability: https://huggingface.co/docs/hub/models-uploading. In this case, the PyTorchModelHubMixin class might be useful given it's a custom PyTorch model.

Linking to the paper can be done by including https://huggingface.co/papers/2404.08351 in the model card.

I see the datasets are already linked: https://huggingface.co/papers/2404.08351, super nice to see :)

mzyyyyyyyy commented 1 week ago

Hi, I will make them available in next weeks. There will be one pretrained model by dataset. As we say in the paper, we pretrain and fine tune on the same dataset and the use of pretrained weights on an other dataset is not trivial.

I am really appreciate it!