├── .gitignore ├── LICENSE.md ├── README.md ├── behavioral_experiments ├── README.md └── human_data.csv ├── configs ├── base.yaml ├── cvrt.yaml └── ssl.yaml ├── data_generation ├── desc.py ├── generalization_tasks.py ├── shape.py ├── tasks.py └── utils.py ├── datasets ├── __init__.py ├── base_datamodules.py ├── cvrt.py ├── cvrt_ssl.py └── transforms_ssl.py ├── generate_dataset.py ├── generate_examples.py ├── inference.py ├── jobs ├── inference.sh ├── train.sh ├── train_array.sh ├── train_comp_ind.sh └── transfer_exp_config.txt ├── main.py ├── misc └── join_dbs.py ├── models ├── __init__.py ├── mocov3.py ├── scheduler.py ├── scn.py ├── vits.py └── wren.py ├── modules ├── __init__.py ├── cnn.py ├── optimizer.py ├── scheduler.py └── ssl.py ├── requirements.txt ├── ssl ├── LICENSE ├── README.md ├── convert_to_deit.py ├── data.py ├── main_moco.py ├── moco │ ├── __init__.py │ ├── builder.py │ ├── loader.py │ └── optimizer.py ├── pretrain_resnet50.sh ├── pretrain_vit_small.sh └── vits.py └── utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/serre-lab/CVR/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/serre-lab/CVR/HEAD/LICENSE.md -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/serre-lab/CVR/HEAD/README.md -------------------------------------------------------------------------------- /behavioral_experiments/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/serre-lab/CVR/HEAD/behavioral_experiments/README.md -------------------------------------------------------------------------------- /behavioral_experiments/human_data.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/serre-lab/CVR/HEAD/behavioral_experiments/human_data.csv -------------------------------------------------------------------------------- /configs/base.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/serre-lab/CVR/HEAD/configs/base.yaml -------------------------------------------------------------------------------- /configs/cvrt.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/serre-lab/CVR/HEAD/configs/cvrt.yaml -------------------------------------------------------------------------------- /configs/ssl.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/serre-lab/CVR/HEAD/configs/ssl.yaml -------------------------------------------------------------------------------- /data_generation/desc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/serre-lab/CVR/HEAD/data_generation/desc.py -------------------------------------------------------------------------------- /data_generation/generalization_tasks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/serre-lab/CVR/HEAD/data_generation/generalization_tasks.py -------------------------------------------------------------------------------- /data_generation/shape.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/serre-lab/CVR/HEAD/data_generation/shape.py -------------------------------------------------------------------------------- /data_generation/tasks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/serre-lab/CVR/HEAD/data_generation/tasks.py -------------------------------------------------------------------------------- /data_generation/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/serre-lab/CVR/HEAD/data_generation/utils.py -------------------------------------------------------------------------------- /datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/serre-lab/CVR/HEAD/datasets/__init__.py -------------------------------------------------------------------------------- /datasets/base_datamodules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/serre-lab/CVR/HEAD/datasets/base_datamodules.py -------------------------------------------------------------------------------- /datasets/cvrt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/serre-lab/CVR/HEAD/datasets/cvrt.py -------------------------------------------------------------------------------- /datasets/cvrt_ssl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/serre-lab/CVR/HEAD/datasets/cvrt_ssl.py -------------------------------------------------------------------------------- /datasets/transforms_ssl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/serre-lab/CVR/HEAD/datasets/transforms_ssl.py -------------------------------------------------------------------------------- /generate_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/serre-lab/CVR/HEAD/generate_dataset.py -------------------------------------------------------------------------------- /generate_examples.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/serre-lab/CVR/HEAD/generate_examples.py -------------------------------------------------------------------------------- /inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/serre-lab/CVR/HEAD/inference.py -------------------------------------------------------------------------------- /jobs/inference.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/serre-lab/CVR/HEAD/jobs/inference.sh -------------------------------------------------------------------------------- /jobs/train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/serre-lab/CVR/HEAD/jobs/train.sh -------------------------------------------------------------------------------- /jobs/train_array.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/serre-lab/CVR/HEAD/jobs/train_array.sh -------------------------------------------------------------------------------- /jobs/train_comp_ind.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/serre-lab/CVR/HEAD/jobs/train_comp_ind.sh -------------------------------------------------------------------------------- /jobs/transfer_exp_config.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/serre-lab/CVR/HEAD/jobs/transfer_exp_config.txt -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/serre-lab/CVR/HEAD/main.py -------------------------------------------------------------------------------- /misc/join_dbs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/serre-lab/CVR/HEAD/misc/join_dbs.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/mocov3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/serre-lab/CVR/HEAD/models/mocov3.py -------------------------------------------------------------------------------- /models/scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/serre-lab/CVR/HEAD/models/scheduler.py -------------------------------------------------------------------------------- /models/scn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/serre-lab/CVR/HEAD/models/scn.py -------------------------------------------------------------------------------- /models/vits.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/serre-lab/CVR/HEAD/models/vits.py -------------------------------------------------------------------------------- /models/wren.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/serre-lab/CVR/HEAD/models/wren.py -------------------------------------------------------------------------------- /modules/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/serre-lab/CVR/HEAD/modules/__init__.py -------------------------------------------------------------------------------- /modules/cnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/serre-lab/CVR/HEAD/modules/cnn.py -------------------------------------------------------------------------------- /modules/optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/serre-lab/CVR/HEAD/modules/optimizer.py -------------------------------------------------------------------------------- /modules/scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/serre-lab/CVR/HEAD/modules/scheduler.py -------------------------------------------------------------------------------- /modules/ssl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/serre-lab/CVR/HEAD/modules/ssl.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/serre-lab/CVR/HEAD/requirements.txt -------------------------------------------------------------------------------- /ssl/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/serre-lab/CVR/HEAD/ssl/LICENSE -------------------------------------------------------------------------------- /ssl/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/serre-lab/CVR/HEAD/ssl/README.md -------------------------------------------------------------------------------- /ssl/convert_to_deit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/serre-lab/CVR/HEAD/ssl/convert_to_deit.py -------------------------------------------------------------------------------- /ssl/data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/serre-lab/CVR/HEAD/ssl/data.py -------------------------------------------------------------------------------- /ssl/main_moco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/serre-lab/CVR/HEAD/ssl/main_moco.py -------------------------------------------------------------------------------- /ssl/moco/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /ssl/moco/builder.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /ssl/moco/loader.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /ssl/moco/optimizer.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /ssl/pretrain_resnet50.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/serre-lab/CVR/HEAD/ssl/pretrain_resnet50.sh -------------------------------------------------------------------------------- /ssl/pretrain_vit_small.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/serre-lab/CVR/HEAD/ssl/pretrain_vit_small.sh -------------------------------------------------------------------------------- /ssl/vits.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/serre-lab/CVR/HEAD/ssl/vits.py -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/serre-lab/CVR/HEAD/utils.py --------------------------------------------------------------------------------