├── .env.example ├── .gitignore ├── .pre-commit-config.yaml ├── .project-root ├── Makefile ├── README.md ├── configs ├── __init__.py ├── callbacks │ ├── default.yaml │ ├── early_stopping.yaml │ ├── model_checkpoint.yaml │ ├── model_summary.yaml │ ├── none.yaml │ └── rich_progress_bar.yaml ├── data │ ├── attributes │ │ ├── 0_CUB_200_2011.yaml │ │ ├── 1_FGVC_AIRCRAFT.yaml │ │ ├── 2_NABirds.yaml │ │ ├── 3_DTD.yaml │ │ ├── 4_OxfordIIITPet.yaml │ │ ├── 5_StanfordDogs.yaml │ │ ├── 6_StanfordCars.yaml │ │ ├── 7_CALTECH101.yaml │ │ ├── 8_CALTECH256.yaml │ │ └── 9_GTSRB.yaml │ └── kd_data.yaml ├── debug │ ├── default.yaml │ ├── fdr.yaml │ ├── limit.yaml │ ├── overfit.yaml │ └── profiler.yaml ├── eval.yaml ├── experiment │ └── example.yaml ├── extras │ └── default.yaml ├── hparams_search │ └── mnist_optuna.yaml ├── hydra │ └── default.yaml ├── local │ └── .gitkeep ├── logger │ ├── aim.yaml │ ├── comet.yaml │ ├── csv.yaml │ ├── many_loggers.yaml │ ├── mlflow.yaml │ ├── neptune.yaml │ ├── tensorboard.yaml │ └── wandb.yaml ├── model │ └── kda.yaml ├── paths │ └── default.yaml ├── train.yaml └── trainer │ ├── cpu.yaml │ ├── ddp.yaml │ ├── ddp_sim.yaml │ ├── default.yaml │ ├── gpu.yaml │ └── mps.yaml ├── environment.yaml ├── notebooks └── .gitkeep ├── pyproject.toml ├── requirements.txt ├── scripts └── schedule.sh ├── setup.py ├── src ├── __init__.py ├── data │ ├── __init__.py │ ├── components │ │ ├── __init__.py │ │ └── kd_dataloader.py │ └── kd_datamodule.py ├── eval.py ├── models │ ├── __init__.py │ ├── components │ │ ├── __init__.py │ │ ├── campus.py │ │ └── criterion.py │ └── kd_module.py ├── train.py ├── train.sh └── utils │ ├── __init__.py │ ├── instantiators.py │ ├── logging_utils.py │ ├── pylogger.py │ ├── rich_utils.py │ └── utils.py └── tests ├── __init__.py ├── conftest.py ├── helpers ├── __init__.py ├── package_available.py ├── run_if.py └── run_sh_command.py ├── test_configs.py ├── test_datamodules.py ├── test_eval.py ├── test_sweeps.py └── test_train.py /.env.example: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsjangAI/VL2Lite/HEAD/.env.example -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsjangAI/VL2Lite/HEAD/.gitignore -------------------------------------------------------------------------------- /.pre-commit-config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsjangAI/VL2Lite/HEAD/.pre-commit-config.yaml -------------------------------------------------------------------------------- /.project-root: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsjangAI/VL2Lite/HEAD/.project-root -------------------------------------------------------------------------------- /Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsjangAI/VL2Lite/HEAD/Makefile -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsjangAI/VL2Lite/HEAD/README.md -------------------------------------------------------------------------------- /configs/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsjangAI/VL2Lite/HEAD/configs/__init__.py -------------------------------------------------------------------------------- /configs/callbacks/default.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsjangAI/VL2Lite/HEAD/configs/callbacks/default.yaml -------------------------------------------------------------------------------- /configs/callbacks/early_stopping.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsjangAI/VL2Lite/HEAD/configs/callbacks/early_stopping.yaml -------------------------------------------------------------------------------- /configs/callbacks/model_checkpoint.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsjangAI/VL2Lite/HEAD/configs/callbacks/model_checkpoint.yaml -------------------------------------------------------------------------------- /configs/callbacks/model_summary.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsjangAI/VL2Lite/HEAD/configs/callbacks/model_summary.yaml -------------------------------------------------------------------------------- /configs/callbacks/none.yaml: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /configs/callbacks/rich_progress_bar.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsjangAI/VL2Lite/HEAD/configs/callbacks/rich_progress_bar.yaml -------------------------------------------------------------------------------- /configs/data/attributes/0_CUB_200_2011.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsjangAI/VL2Lite/HEAD/configs/data/attributes/0_CUB_200_2011.yaml -------------------------------------------------------------------------------- /configs/data/attributes/1_FGVC_AIRCRAFT.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsjangAI/VL2Lite/HEAD/configs/data/attributes/1_FGVC_AIRCRAFT.yaml -------------------------------------------------------------------------------- /configs/data/attributes/2_NABirds.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsjangAI/VL2Lite/HEAD/configs/data/attributes/2_NABirds.yaml -------------------------------------------------------------------------------- /configs/data/attributes/3_DTD.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsjangAI/VL2Lite/HEAD/configs/data/attributes/3_DTD.yaml -------------------------------------------------------------------------------- /configs/data/attributes/4_OxfordIIITPet.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsjangAI/VL2Lite/HEAD/configs/data/attributes/4_OxfordIIITPet.yaml -------------------------------------------------------------------------------- /configs/data/attributes/5_StanfordDogs.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsjangAI/VL2Lite/HEAD/configs/data/attributes/5_StanfordDogs.yaml -------------------------------------------------------------------------------- /configs/data/attributes/6_StanfordCars.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsjangAI/VL2Lite/HEAD/configs/data/attributes/6_StanfordCars.yaml -------------------------------------------------------------------------------- /configs/data/attributes/7_CALTECH101.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsjangAI/VL2Lite/HEAD/configs/data/attributes/7_CALTECH101.yaml -------------------------------------------------------------------------------- /configs/data/attributes/8_CALTECH256.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsjangAI/VL2Lite/HEAD/configs/data/attributes/8_CALTECH256.yaml -------------------------------------------------------------------------------- /configs/data/attributes/9_GTSRB.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsjangAI/VL2Lite/HEAD/configs/data/attributes/9_GTSRB.yaml -------------------------------------------------------------------------------- /configs/data/kd_data.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsjangAI/VL2Lite/HEAD/configs/data/kd_data.yaml -------------------------------------------------------------------------------- /configs/debug/default.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsjangAI/VL2Lite/HEAD/configs/debug/default.yaml -------------------------------------------------------------------------------- /configs/debug/fdr.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsjangAI/VL2Lite/HEAD/configs/debug/fdr.yaml -------------------------------------------------------------------------------- /configs/debug/limit.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsjangAI/VL2Lite/HEAD/configs/debug/limit.yaml -------------------------------------------------------------------------------- /configs/debug/overfit.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsjangAI/VL2Lite/HEAD/configs/debug/overfit.yaml -------------------------------------------------------------------------------- /configs/debug/profiler.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsjangAI/VL2Lite/HEAD/configs/debug/profiler.yaml -------------------------------------------------------------------------------- /configs/eval.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsjangAI/VL2Lite/HEAD/configs/eval.yaml -------------------------------------------------------------------------------- /configs/experiment/example.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsjangAI/VL2Lite/HEAD/configs/experiment/example.yaml -------------------------------------------------------------------------------- /configs/extras/default.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsjangAI/VL2Lite/HEAD/configs/extras/default.yaml -------------------------------------------------------------------------------- /configs/hparams_search/mnist_optuna.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsjangAI/VL2Lite/HEAD/configs/hparams_search/mnist_optuna.yaml -------------------------------------------------------------------------------- /configs/hydra/default.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsjangAI/VL2Lite/HEAD/configs/hydra/default.yaml -------------------------------------------------------------------------------- /configs/local/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /configs/logger/aim.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsjangAI/VL2Lite/HEAD/configs/logger/aim.yaml -------------------------------------------------------------------------------- /configs/logger/comet.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsjangAI/VL2Lite/HEAD/configs/logger/comet.yaml -------------------------------------------------------------------------------- /configs/logger/csv.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsjangAI/VL2Lite/HEAD/configs/logger/csv.yaml -------------------------------------------------------------------------------- /configs/logger/many_loggers.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsjangAI/VL2Lite/HEAD/configs/logger/many_loggers.yaml -------------------------------------------------------------------------------- /configs/logger/mlflow.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsjangAI/VL2Lite/HEAD/configs/logger/mlflow.yaml -------------------------------------------------------------------------------- /configs/logger/neptune.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsjangAI/VL2Lite/HEAD/configs/logger/neptune.yaml -------------------------------------------------------------------------------- /configs/logger/tensorboard.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsjangAI/VL2Lite/HEAD/configs/logger/tensorboard.yaml -------------------------------------------------------------------------------- /configs/logger/wandb.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsjangAI/VL2Lite/HEAD/configs/logger/wandb.yaml -------------------------------------------------------------------------------- /configs/model/kda.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsjangAI/VL2Lite/HEAD/configs/model/kda.yaml -------------------------------------------------------------------------------- /configs/paths/default.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsjangAI/VL2Lite/HEAD/configs/paths/default.yaml -------------------------------------------------------------------------------- /configs/train.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsjangAI/VL2Lite/HEAD/configs/train.yaml -------------------------------------------------------------------------------- /configs/trainer/cpu.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsjangAI/VL2Lite/HEAD/configs/trainer/cpu.yaml -------------------------------------------------------------------------------- /configs/trainer/ddp.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsjangAI/VL2Lite/HEAD/configs/trainer/ddp.yaml -------------------------------------------------------------------------------- /configs/trainer/ddp_sim.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsjangAI/VL2Lite/HEAD/configs/trainer/ddp_sim.yaml -------------------------------------------------------------------------------- /configs/trainer/default.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsjangAI/VL2Lite/HEAD/configs/trainer/default.yaml -------------------------------------------------------------------------------- /configs/trainer/gpu.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsjangAI/VL2Lite/HEAD/configs/trainer/gpu.yaml -------------------------------------------------------------------------------- /configs/trainer/mps.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsjangAI/VL2Lite/HEAD/configs/trainer/mps.yaml -------------------------------------------------------------------------------- /environment.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsjangAI/VL2Lite/HEAD/environment.yaml -------------------------------------------------------------------------------- /notebooks/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsjangAI/VL2Lite/HEAD/pyproject.toml -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsjangAI/VL2Lite/HEAD/requirements.txt -------------------------------------------------------------------------------- /scripts/schedule.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsjangAI/VL2Lite/HEAD/scripts/schedule.sh -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsjangAI/VL2Lite/HEAD/setup.py -------------------------------------------------------------------------------- /src/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/data/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/data/components/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/data/components/kd_dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsjangAI/VL2Lite/HEAD/src/data/components/kd_dataloader.py -------------------------------------------------------------------------------- /src/data/kd_datamodule.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsjangAI/VL2Lite/HEAD/src/data/kd_datamodule.py -------------------------------------------------------------------------------- /src/eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsjangAI/VL2Lite/HEAD/src/eval.py -------------------------------------------------------------------------------- /src/models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/models/components/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/models/components/campus.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsjangAI/VL2Lite/HEAD/src/models/components/campus.py -------------------------------------------------------------------------------- /src/models/components/criterion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsjangAI/VL2Lite/HEAD/src/models/components/criterion.py -------------------------------------------------------------------------------- /src/models/kd_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsjangAI/VL2Lite/HEAD/src/models/kd_module.py -------------------------------------------------------------------------------- /src/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsjangAI/VL2Lite/HEAD/src/train.py -------------------------------------------------------------------------------- /src/train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsjangAI/VL2Lite/HEAD/src/train.sh -------------------------------------------------------------------------------- /src/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsjangAI/VL2Lite/HEAD/src/utils/__init__.py -------------------------------------------------------------------------------- /src/utils/instantiators.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsjangAI/VL2Lite/HEAD/src/utils/instantiators.py -------------------------------------------------------------------------------- /src/utils/logging_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsjangAI/VL2Lite/HEAD/src/utils/logging_utils.py -------------------------------------------------------------------------------- /src/utils/pylogger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsjangAI/VL2Lite/HEAD/src/utils/pylogger.py -------------------------------------------------------------------------------- /src/utils/rich_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsjangAI/VL2Lite/HEAD/src/utils/rich_utils.py -------------------------------------------------------------------------------- /src/utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsjangAI/VL2Lite/HEAD/src/utils/utils.py -------------------------------------------------------------------------------- /tests/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/conftest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsjangAI/VL2Lite/HEAD/tests/conftest.py -------------------------------------------------------------------------------- /tests/helpers/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/helpers/package_available.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsjangAI/VL2Lite/HEAD/tests/helpers/package_available.py -------------------------------------------------------------------------------- /tests/helpers/run_if.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsjangAI/VL2Lite/HEAD/tests/helpers/run_if.py -------------------------------------------------------------------------------- /tests/helpers/run_sh_command.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsjangAI/VL2Lite/HEAD/tests/helpers/run_sh_command.py -------------------------------------------------------------------------------- /tests/test_configs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsjangAI/VL2Lite/HEAD/tests/test_configs.py -------------------------------------------------------------------------------- /tests/test_datamodules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsjangAI/VL2Lite/HEAD/tests/test_datamodules.py -------------------------------------------------------------------------------- /tests/test_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsjangAI/VL2Lite/HEAD/tests/test_eval.py -------------------------------------------------------------------------------- /tests/test_sweeps.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsjangAI/VL2Lite/HEAD/tests/test_sweeps.py -------------------------------------------------------------------------------- /tests/test_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsjangAI/VL2Lite/HEAD/tests/test_train.py --------------------------------------------------------------------------------