Commit Graph

33 Commits

Author SHA1 Message Date
Yann Ahlgrim 99489c52dd remove params overwrite 2026-06-19 11:24:08 +02:00
Yann Ahlgrim 2308540971 add tag to folder name + change grid config yaml 2026-06-19 10:18:16 +02:00
Yann Ahlgrim 918c1597bc gradient checkpointing 2026-06-18 16:37:14 +02:00
Yann Ahlgrim 772acd8f0d gradient accumulation 2026-06-18 16:25:52 +02:00
Yann Ahlgrim ccdddc47e5 bug: fixed port 2026-06-09 01:07:32 +02:00
Yann Ahlgrim ca026ffecb gradient accumulation 2026-06-09 00:58:25 +02:00
YannAhlgrim 45269d16bb gradient checkpointing 2026-06-04 14:09:14 +02:00
YannAhlgrim f2b501d8c9 adapt params to ijepa paper 2026-06-02 12:33:51 +02:00
YannAhlgrim 150c2a7253 fix: do not run 4 times and log new params 2026-06-02 12:00:12 +02:00
YannAhlgrim 2556c17d88 remove mlp + add batch norm + last 4 avg pool layer as in the paper of IJEPA 2026-06-02 11:47:21 +02:00
YannAhlgrim a9599c7220 bug: missing fields 2026-05-27 17:08:58 +02:00
Yann Ahlgrim 6f5c6b8f26 del folder afterwards 2026-05-25 16:52:18 +02:00
YannAhlgrim 825281b65a mlp layer + new grid 2026-05-24 09:46:54 +02:00
YannAhlgrim 882c2b53b6 adapt grid 2026-05-23 13:54:54 +02:00
YannAhlgrim 34a15d646d add grid search for supervised pipeline 2026-05-23 13:40:46 +02:00
YannAhlgrim 7e25f2a57f auto_folder 2026-05-22 16:08:47 +02:00
YannAhlgrim 5ce42b8ad0 add lars opt and configs 2026-05-20 15:05:18 +02:00
YannAhlgrim e0bd9fc7ce import dist 2026-05-16 12:47:41 +02:00
YannAhlgrim 79fc9564ea DDP guard if single GPUgio 2026-05-16 11:52:10 +02:00
YannAhlgrim 4df162180e disable random resize crop + ft config 2026-05-16 11:04:00 +02:00
YannAhlgrim d22a3aba90 evaluation pipeline 2026-05-14 10:52:40 +02:00
YannAhlgrim 75cf127d74 add checkpoint_folder + tweak params for supervised fine tuning + remove default values of params 2026-04-27 11:16:26 +02:00
YannAhlgrim 459d998b20 add early stopping callback 2026-04-25 11:03:05 +02:00
YannAhlgrim c54816aa93 add evaluation transform method 2026-04-25 11:01:38 +02:00
YannAhlgrim 02991a7eb5 no grad if frozen in forward pass 2026-04-25 11:00:14 +02:00
ahyanc25@reutlingen-university.de 417319457a remove image folder 2026-04-25 10:14:22 +02:00
YannAhlgrim 7e32fc994f return tuple (img, target) even on unlabeled data 2026-04-18 15:02:17 +02:00
YannAhlgrim 4c4d6713e0 add save checkpoint logic 2026-04-15 13:43:05 +02:00
YannAhlgrim 0a38a264f7 add supervised training pipeline 2026-04-15 13:22:44 +02:00
YannAhlgrim e2910cb458 Header for Linear Probing 2026-04-12 11:19:06 +02:00
YannAhlgrim 4d2fd1021b integrate make_iwildcam in train.py 2026-04-02 14:15:06 +02:00
YannAhlgrim 5165607f55 wilds dataset loader 2026-04-01 16:26:35 +02:00
Mido Assran 52c1ae95d0 Initial commit 2023-06-13 13:03:30 +00:00