Training the JEPA Model

To train the JEPA model, run the following command:

python best_model.py

Evaluating the JEPA Model

To evaluate the JEPA model on the four downstream tasks, run the following command:

python main.py

Experiment Results

See: https://docs.google.com/spreadsheets/d/1p8ctj7APCc-4V9X6_7Hehxh6ywjLobTBXt9OVpwn1ck/edit?usp=sharing