YannAhlgrim 45269d16bb gradient checkpointing

2026-06-04 14:09:14 +02:00

..

datasets

evaluation pipeline

2026-05-14 10:52:40 +02:00

masks

Initial commit

2023-06-13 13:03:30 +00:00

models

gradient checkpointing

2026-06-04 14:09:14 +02:00

utils

adapt params to ijepa paper

2026-06-02 12:33:51 +02:00

eval_wilds.py

remove mlp + add batch norm + last 4 avg pool layer as in the paper of IJEPA

2026-06-02 11:47:21 +02:00

helper.py

gradient checkpointing

2026-06-04 14:09:14 +02:00

train_supervised.py

fix: do not run 4 times and log new params

2026-06-02 12:00:12 +02:00

train.py

gradient checkpointing

2026-06-04 14:09:14 +02:00

transforms.py

disable random resize crop + ft config

2026-05-16 11:04:00 +02:00