Code

Add IJEPA task#25

Code

Merged

fcogidi merged 35 commits intomainVectorInstitute/mmlearn:mainfrom ijepa_trainingVectorInstitute/mmlearn:ijepa_trainingCopy head branch name to clipboardDec 5, 2024

+761-193

Commits

Commits on Oct 9, 2024

• add v1 of ijepa task

  vahid0001

  committed
  e108aef

  Copy full SHA for e108aef

• reorganize imports

  vahid0001

  committed
  763a7fa

  Copy full SHA for 763a7fa

• fix precommits

  vahid0001

  committed
  cc98d33

  Copy full SHA for cc98d33

• change docstring

  vahid0001

  committed
  f7fdcea

  Copy full SHA for f7fdcea

Commits on Oct 10, 2024

• revise optimizer configure method

  vahid0001

  committed
  be3ac26

  Copy full SHA for be3ac26

Commits on Oct 16, 2024

• add load checkpoint to vit class

  vahid0001

  committed
  a1dcd84

  Copy full SHA for a1dcd84

• revise ijepa class

  vahid0001

  committed
  41a21f6

  Copy full SHA for 41a21f6

Commits on Oct 30, 2024

• remove load checkpoint

  vahid0001

  committed
  f9ee90a

  Copy full SHA for f9ee90a

• revise ijepa trainer class

  vahid0001

  committed
  8d41a76

  Copy full SHA for 8d41a76

• change/add files for ijepa training

  vahid0001

  committed
  b671d09

  Copy full SHA for b671d09

Commits on Nov 19, 2024

• add IJEPA task

  vahid0001

  committed
  0f44546

  Copy full SHA for 0f44546

• revise masking function

  vahid0001

  committed
  b91b9ea

  Copy full SHA for b91b9ea

• revise vit classes

  vahid0001

  committed
  d4be471

  Copy full SHA for d4be471

• add IJEPA task

  vahid0001

  committed
  9046cbb

  Copy full SHA for 9046cbb

• revise masking function

  vahid0001

  committed
  4cd057c

  Copy full SHA for 4cd057c

• revise vit classes

  vahid0001

  committed
  48eea90

  Copy full SHA for 48eea90

• remove state_dict method

  vahid0001

  committed
  6db4155

  Copy full SHA for 6db4155

• revise ijepa pretraining task

  vahid0001

  committed
  dbc4154

  Copy full SHA for dbc4154

• revise ijepa config file

  vahid0001

  committed
  583c020

  Copy full SHA for 583c020

• add init to ijepa project config file

  vahid0001

  committed
  871e6a2

  Copy full SHA for 871e6a2

Commits on Nov 21, 2024

• add proper transformations

  vahid0001

  committed
  f508293

  Copy full SHA for f508293

• revise ijepa config file

  vahid0001

  committed
  336c040

  Copy full SHA for 336c040

• change some hp, fix precommit errors

  vahid0001

  committed
  c06390c

  Copy full SHA for c06390c

• fix precommit errors

  vahid0001

  committed
  facac4b

  Copy full SHA for facac4b

• fix type annotation

  vahid0001

  committed
  6e91766

  Copy full SHA for 6e91766

Commits on Nov 27, 2024

• Remove IJEPA import from __init__.py and add it to tasks/__init__.py

  fcogidi

  committed
  2fe2a7c

  Copy full SHA for 2fe2a7c

• Refactor ExponentialMovingAverage class in ema.py

  fcogidi

  committed
  7a0bd33

  Copy full SHA for 7a0bd33

• Add base class for tasks that require training

  fcogidi

  committed
  70bd0e6

  Copy full SHA for 70bd0e6

• Update configuration file for reproducing ImageNet experiment

  fcogidi

  committed
  b0be82b

  Copy full SHA for b0be82b

• Merge branch 'main' into ijepa_training

  fcogidi

  authored
  e37d476

  Copy full SHA for e37d476

• Fix modalities key in ijepa_pretraining.py

  fcogidi

  committed
  099ae33

  Copy full SHA for 099ae33

• Fix type hints and rename ijepa task module

  fcogidi

  committed
  39c2e19

  Copy full SHA for 39c2e19

• Add dirpath for saving checkpoints in Vector SLURM environment

  fcogidi

  committed
  5e19f87

  Copy full SHA for 5e19f87

• Fix import statement for IJEPA class

  fcogidi

  committed
  5937565

  Copy full SHA for 5937565

Commits on Dec 5, 2024

• Merge branch 'main' into ijepa_training

  fcogidi

  committed
  ceed49d

  Copy full SHA for ceed49d