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

Datamodule & Model for Clay v1 #253

Merged
merged 13 commits into from
May 24, 2024
Merged

Datamodule & Model for Clay v1 #253

merged 13 commits into from
May 24, 2024

Conversation

srmsoumya
Copy link
Collaborator

  • Add support batch loading of data from different sensors
  • Dynamic Embeddings to patchify chips from different sensors
  • Add teacher model to better align the model embeddings
  • Add support to feed GSD & time/latlon encodings

srmsoumya and others added 11 commits April 18, 2024 21:25
- Add starter code for clay v1 model
- Add `model_clay_v1.py` that handles images of different size & spectral bands
- Add `metadata.yaml` that contains meta information of satellite data, like wavelengths & gsd
- Add cls tokens to ClayMAE
- Use lightning config file
- Dynamic Embedding for model
- Use DINO as a teacher to learn model embeddings
- Support multi-platform model training using custom samplers & data cubes
@srmsoumya srmsoumya requested a review from yellowcap May 24, 2024 06:04
Copy link
Member

@yellowcap yellowcap left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Here goes the model! 🐠

@srmsoumya srmsoumya merged commit 34e8b36 into main May 24, 2024
3 checks passed
@srmsoumya srmsoumya deleted the dev branch May 24, 2024 08:50
@yellowcap yellowcap mentioned this pull request May 24, 2024
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

Successfully merging this pull request may close these issues.

2 participants