search

gaasher / I-JEPA

Implementation of I-JEPA from "Self-Supervised Learning from Images with a Joint-Embedding Predictive Architecture"
MIT License
254 stars 26 forks source link

No train.py file #8

Closed Sagor-Saha closed 1 year ago

Sagor-Saha commented 1 year ago

In order to run, just run: python train.py on your command line Can you please add train.py file?

gaasher commented 1 year ago

Pretrain with pretrain_IJEPA.py and finetune wiht finetune_IJEPA.py. You will have to adjust the dataset classes to support your images (i.e loading directly with opencv/zarr/one really big dataframe)