gaasher/I-JEPA

No train.py file

Closed this issue · 1 comments

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

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)