├── .dockerignore ├── .gitattributes ├── .gitignore ├── LICENSE ├── README.md ├── cache_creation.py ├── checkpoints └── checkpoints.txt ├── data ├── README.md ├── cub │ ├── .gitignore │ └── split │ │ ├── test.csv │ │ ├── train.csv │ │ └── val.csv └── miniimagenet │ ├── .gitignore │ └── split │ ├── class_names.csv │ ├── test.csv │ ├── train.csv │ └── val.csv ├── docker ├── Dockerfile ├── docker-compose.yml └── y ├── docker_nvidia ├── Dockerfile ├── docker-compose.yml └── y ├── download_weight.sh ├── embeds_cache └── embeds_cache.txt ├── exp_results.txt ├── imgs ├── base_illustrative_centaur_new.png ├── cub.png ├── mini.png └── tiered.png ├── model ├── __init__.py ├── data_parallel.py ├── dataloader │ ├── cub.py │ ├── mini_imagenet.py │ ├── samplers.py │ ├── simclr_view_generator.py │ ├── tiered_imagenet.py │ └── tiered_imagenet_og.py ├── logger.py ├── models │ ├── __init__.py │ ├── base.py │ ├── bilstm.py │ ├── bit3_pretrain.py │ ├── classifier.py │ ├── deepset.py │ ├── feat.py │ ├── feat_aux.py │ ├── feat_base_salcrop.py │ ├── feat_baseclass.py │ ├── feat_baseclass2.py │ ├── feat_basetransformer.py │ ├── feat_basetransformer2.py │ ├── feat_basetransformer3.py │ ├── feat_basetransformer3_2d.py │ ├── feat_basetransformer3_2d_5shot.py │ ├── feat_basetransformer3_2d_bencoder.py │ ├── feat_basetransformer3_2d_patch.py │ ├── feat_basetransformer3_2d_sym.py │ ├── feat_basetransformer3_aux.py │ ├── feat_sal.py │ ├── feat_sal2.py │ ├── featstar.py │ ├── featv3_1.py │ ├── featv3_2_2.py │ ├── featv3_3.py │ ├── graphnet.py │ ├── matchnet.py │ ├── protonet.py │ ├── semi_feat.py │ └── semi_protofeat.py ├── networks │ ├── WRN28.py │ ├── convnet.py │ ├── dropblock.py │ ├── res12.py │ └── res18.py ├── trainer │ ├── __init__.py │ ├── base.py │ ├── fsl_trainer.py │ └── helpers.py └── utils.py ├── nb_utils ├── cache_utils.py ├── imagenet_classes.json ├── prototype_utils.py ├── sal_utils.py ├── vis_utils.py └── wnids.json ├── pretrain.py ├── pretrain_l2l.py ├── pretrain_patches.py ├── pretrain_simclr.py ├── pretrain_transformer.py ├── saves └── saves.txt ├── scheduler.py ├── train_fsl.py └── training_tasks.json /.dockerignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mayug/BaseTransformers/HEAD/.dockerignore -------------------------------------------------------------------------------- /.gitattributes: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mayug/BaseTransformers/HEAD/.gitattributes -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mayug/BaseTransformers/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mayug/BaseTransformers/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mayug/BaseTransformers/HEAD/README.md -------------------------------------------------------------------------------- /cache_creation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mayug/BaseTransformers/HEAD/cache_creation.py -------------------------------------------------------------------------------- /checkpoints/checkpoints.txt: -------------------------------------------------------------------------------- 1 | # This folder saves all the logs,checkpoints,wandb_logs. 2 | -------------------------------------------------------------------------------- /data/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mayug/BaseTransformers/HEAD/data/README.md -------------------------------------------------------------------------------- /data/cub/.gitignore: -------------------------------------------------------------------------------- 1 | images 2 | -------------------------------------------------------------------------------- /data/cub/split/test.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mayug/BaseTransformers/HEAD/data/cub/split/test.csv -------------------------------------------------------------------------------- /data/cub/split/train.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mayug/BaseTransformers/HEAD/data/cub/split/train.csv -------------------------------------------------------------------------------- /data/cub/split/val.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mayug/BaseTransformers/HEAD/data/cub/split/val.csv -------------------------------------------------------------------------------- /data/miniimagenet/.gitignore: -------------------------------------------------------------------------------- 1 | images 2 | -------------------------------------------------------------------------------- /data/miniimagenet/split/class_names.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mayug/BaseTransformers/HEAD/data/miniimagenet/split/class_names.csv -------------------------------------------------------------------------------- /data/miniimagenet/split/test.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mayug/BaseTransformers/HEAD/data/miniimagenet/split/test.csv -------------------------------------------------------------------------------- /data/miniimagenet/split/train.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mayug/BaseTransformers/HEAD/data/miniimagenet/split/train.csv -------------------------------------------------------------------------------- /data/miniimagenet/split/val.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mayug/BaseTransformers/HEAD/data/miniimagenet/split/val.csv -------------------------------------------------------------------------------- /docker/Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mayug/BaseTransformers/HEAD/docker/Dockerfile -------------------------------------------------------------------------------- /docker/docker-compose.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mayug/BaseTransformers/HEAD/docker/docker-compose.yml -------------------------------------------------------------------------------- /docker/y: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mayug/BaseTransformers/HEAD/docker/y -------------------------------------------------------------------------------- /docker_nvidia/Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mayug/BaseTransformers/HEAD/docker_nvidia/Dockerfile -------------------------------------------------------------------------------- /docker_nvidia/docker-compose.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mayug/BaseTransformers/HEAD/docker_nvidia/docker-compose.yml -------------------------------------------------------------------------------- /docker_nvidia/y: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mayug/BaseTransformers/HEAD/docker_nvidia/y -------------------------------------------------------------------------------- /download_weight.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mayug/BaseTransformers/HEAD/download_weight.sh -------------------------------------------------------------------------------- /embeds_cache/embeds_cache.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mayug/BaseTransformers/HEAD/embeds_cache/embeds_cache.txt -------------------------------------------------------------------------------- /exp_results.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mayug/BaseTransformers/HEAD/exp_results.txt -------------------------------------------------------------------------------- /imgs/base_illustrative_centaur_new.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mayug/BaseTransformers/HEAD/imgs/base_illustrative_centaur_new.png -------------------------------------------------------------------------------- /imgs/cub.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mayug/BaseTransformers/HEAD/imgs/cub.png -------------------------------------------------------------------------------- /imgs/mini.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mayug/BaseTransformers/HEAD/imgs/mini.png -------------------------------------------------------------------------------- /imgs/tiered.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mayug/BaseTransformers/HEAD/imgs/tiered.png -------------------------------------------------------------------------------- /model/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /model/data_parallel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mayug/BaseTransformers/HEAD/model/data_parallel.py -------------------------------------------------------------------------------- /model/dataloader/cub.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mayug/BaseTransformers/HEAD/model/dataloader/cub.py -------------------------------------------------------------------------------- /model/dataloader/mini_imagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mayug/BaseTransformers/HEAD/model/dataloader/mini_imagenet.py -------------------------------------------------------------------------------- /model/dataloader/samplers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mayug/BaseTransformers/HEAD/model/dataloader/samplers.py -------------------------------------------------------------------------------- /model/dataloader/simclr_view_generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mayug/BaseTransformers/HEAD/model/dataloader/simclr_view_generator.py -------------------------------------------------------------------------------- /model/dataloader/tiered_imagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mayug/BaseTransformers/HEAD/model/dataloader/tiered_imagenet.py -------------------------------------------------------------------------------- /model/dataloader/tiered_imagenet_og.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mayug/BaseTransformers/HEAD/model/dataloader/tiered_imagenet_og.py -------------------------------------------------------------------------------- /model/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mayug/BaseTransformers/HEAD/model/logger.py -------------------------------------------------------------------------------- /model/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mayug/BaseTransformers/HEAD/model/models/__init__.py -------------------------------------------------------------------------------- /model/models/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mayug/BaseTransformers/HEAD/model/models/base.py -------------------------------------------------------------------------------- /model/models/bilstm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mayug/BaseTransformers/HEAD/model/models/bilstm.py -------------------------------------------------------------------------------- /model/models/bit3_pretrain.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mayug/BaseTransformers/HEAD/model/models/bit3_pretrain.py -------------------------------------------------------------------------------- /model/models/classifier.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mayug/BaseTransformers/HEAD/model/models/classifier.py -------------------------------------------------------------------------------- /model/models/deepset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mayug/BaseTransformers/HEAD/model/models/deepset.py -------------------------------------------------------------------------------- /model/models/feat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mayug/BaseTransformers/HEAD/model/models/feat.py -------------------------------------------------------------------------------- /model/models/feat_aux.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mayug/BaseTransformers/HEAD/model/models/feat_aux.py -------------------------------------------------------------------------------- /model/models/feat_base_salcrop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mayug/BaseTransformers/HEAD/model/models/feat_base_salcrop.py -------------------------------------------------------------------------------- /model/models/feat_baseclass.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mayug/BaseTransformers/HEAD/model/models/feat_baseclass.py -------------------------------------------------------------------------------- /model/models/feat_baseclass2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mayug/BaseTransformers/HEAD/model/models/feat_baseclass2.py -------------------------------------------------------------------------------- /model/models/feat_basetransformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mayug/BaseTransformers/HEAD/model/models/feat_basetransformer.py -------------------------------------------------------------------------------- /model/models/feat_basetransformer2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mayug/BaseTransformers/HEAD/model/models/feat_basetransformer2.py -------------------------------------------------------------------------------- /model/models/feat_basetransformer3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mayug/BaseTransformers/HEAD/model/models/feat_basetransformer3.py -------------------------------------------------------------------------------- /model/models/feat_basetransformer3_2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mayug/BaseTransformers/HEAD/model/models/feat_basetransformer3_2d.py -------------------------------------------------------------------------------- /model/models/feat_basetransformer3_2d_5shot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mayug/BaseTransformers/HEAD/model/models/feat_basetransformer3_2d_5shot.py -------------------------------------------------------------------------------- /model/models/feat_basetransformer3_2d_bencoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mayug/BaseTransformers/HEAD/model/models/feat_basetransformer3_2d_bencoder.py -------------------------------------------------------------------------------- /model/models/feat_basetransformer3_2d_patch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mayug/BaseTransformers/HEAD/model/models/feat_basetransformer3_2d_patch.py -------------------------------------------------------------------------------- /model/models/feat_basetransformer3_2d_sym.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mayug/BaseTransformers/HEAD/model/models/feat_basetransformer3_2d_sym.py -------------------------------------------------------------------------------- /model/models/feat_basetransformer3_aux.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mayug/BaseTransformers/HEAD/model/models/feat_basetransformer3_aux.py -------------------------------------------------------------------------------- /model/models/feat_sal.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mayug/BaseTransformers/HEAD/model/models/feat_sal.py -------------------------------------------------------------------------------- /model/models/feat_sal2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mayug/BaseTransformers/HEAD/model/models/feat_sal2.py -------------------------------------------------------------------------------- /model/models/featstar.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mayug/BaseTransformers/HEAD/model/models/featstar.py -------------------------------------------------------------------------------- /model/models/featv3_1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mayug/BaseTransformers/HEAD/model/models/featv3_1.py -------------------------------------------------------------------------------- /model/models/featv3_2_2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mayug/BaseTransformers/HEAD/model/models/featv3_2_2.py -------------------------------------------------------------------------------- /model/models/featv3_3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mayug/BaseTransformers/HEAD/model/models/featv3_3.py -------------------------------------------------------------------------------- /model/models/graphnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mayug/BaseTransformers/HEAD/model/models/graphnet.py -------------------------------------------------------------------------------- /model/models/matchnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mayug/BaseTransformers/HEAD/model/models/matchnet.py -------------------------------------------------------------------------------- /model/models/protonet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mayug/BaseTransformers/HEAD/model/models/protonet.py -------------------------------------------------------------------------------- /model/models/semi_feat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mayug/BaseTransformers/HEAD/model/models/semi_feat.py -------------------------------------------------------------------------------- /model/models/semi_protofeat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mayug/BaseTransformers/HEAD/model/models/semi_protofeat.py -------------------------------------------------------------------------------- /model/networks/WRN28.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mayug/BaseTransformers/HEAD/model/networks/WRN28.py -------------------------------------------------------------------------------- /model/networks/convnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mayug/BaseTransformers/HEAD/model/networks/convnet.py -------------------------------------------------------------------------------- /model/networks/dropblock.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mayug/BaseTransformers/HEAD/model/networks/dropblock.py -------------------------------------------------------------------------------- /model/networks/res12.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mayug/BaseTransformers/HEAD/model/networks/res12.py -------------------------------------------------------------------------------- /model/networks/res18.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mayug/BaseTransformers/HEAD/model/networks/res18.py -------------------------------------------------------------------------------- /model/trainer/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /model/trainer/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mayug/BaseTransformers/HEAD/model/trainer/base.py -------------------------------------------------------------------------------- /model/trainer/fsl_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mayug/BaseTransformers/HEAD/model/trainer/fsl_trainer.py -------------------------------------------------------------------------------- /model/trainer/helpers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mayug/BaseTransformers/HEAD/model/trainer/helpers.py -------------------------------------------------------------------------------- /model/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mayug/BaseTransformers/HEAD/model/utils.py -------------------------------------------------------------------------------- /nb_utils/cache_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mayug/BaseTransformers/HEAD/nb_utils/cache_utils.py -------------------------------------------------------------------------------- /nb_utils/imagenet_classes.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mayug/BaseTransformers/HEAD/nb_utils/imagenet_classes.json -------------------------------------------------------------------------------- /nb_utils/prototype_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mayug/BaseTransformers/HEAD/nb_utils/prototype_utils.py -------------------------------------------------------------------------------- /nb_utils/sal_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mayug/BaseTransformers/HEAD/nb_utils/sal_utils.py -------------------------------------------------------------------------------- /nb_utils/vis_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mayug/BaseTransformers/HEAD/nb_utils/vis_utils.py -------------------------------------------------------------------------------- /nb_utils/wnids.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mayug/BaseTransformers/HEAD/nb_utils/wnids.json -------------------------------------------------------------------------------- /pretrain.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mayug/BaseTransformers/HEAD/pretrain.py -------------------------------------------------------------------------------- /pretrain_l2l.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mayug/BaseTransformers/HEAD/pretrain_l2l.py -------------------------------------------------------------------------------- /pretrain_patches.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mayug/BaseTransformers/HEAD/pretrain_patches.py -------------------------------------------------------------------------------- /pretrain_simclr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mayug/BaseTransformers/HEAD/pretrain_simclr.py -------------------------------------------------------------------------------- /pretrain_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mayug/BaseTransformers/HEAD/pretrain_transformer.py -------------------------------------------------------------------------------- /saves/saves.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mayug/BaseTransformers/HEAD/saves/saves.txt -------------------------------------------------------------------------------- /scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mayug/BaseTransformers/HEAD/scheduler.py -------------------------------------------------------------------------------- /train_fsl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mayug/BaseTransformers/HEAD/train_fsl.py -------------------------------------------------------------------------------- /training_tasks.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mayug/BaseTransformers/HEAD/training_tasks.json --------------------------------------------------------------------------------