Classifier downstream task

Clay-foundation / model

The Clay Foundation Model (in development)

https://clay-foundation.github.io/model/

Apache License 2.0

236 stars 25 forks source link

Closed srmsoumya closed 2 weeks ago

srmsoumya commented 4 weeks ago

Add a classifier downstream task using Clay Encoder.

In this PR

Download & load Eurosat dataset from TorchGeo
Train just the model head for 20 epochs
Report the benchmark results in https://github.com/Clay-foundation/model/discussions/269