├── LICENSE ├── README.md ├── dataset_configs ├── CIFARFS.yaml.example ├── miniimagenet.yaml.example └── tieredimagenet.yaml.example ├── environment.pytorch13.yml ├── expm_args └── .gitignore ├── results └── .gitignore ├── runs └── .gitignore ├── scripts ├── bash_scripts │ ├── abl_all_cifar_proto.sh │ ├── abl_all_miniimagenet_proto.sh │ ├── abl_nca_fixed_batch_composition.sh │ ├── abl_nca_fixed_batch_composition_128.sh │ ├── abl_nca_fixed_batch_composition_512.sh │ ├── abl_nca_repl_sampling.sh │ ├── abl_nca_repl_sampling_128.sh │ ├── abl_nca_repl_sampling_512.sh │ ├── abl_proto_no_SQ_separation.sh │ ├── abl_proto_no_SQ_separation_128.sh │ ├── abl_proto_no_SQ_separation_512.sh │ ├── abl_proto_no_aggregates.sh │ ├── abl_proto_no_aggregates_128.sh │ ├── abl_proto_no_aggregates_512.sh │ ├── batch_expm_cifar_matching.sh │ ├── batch_expm_cifar_proto.sh │ ├── batch_expm_miniimagenet_matching.sh │ ├── batch_expm_miniimagenet_proto.sh │ ├── generic_nca.sh │ ├── generic_nca_matching.sh │ ├── generic_proto.sh │ ├── generic_proto_matching.sh │ ├── matching_nets_ablation_example.sh │ ├── matchingnew_cifar.sh │ ├── matchingnew_rn12_tiered.sh │ ├── matchingold_rn12_cifar.sh │ ├── matchingold_rn12_miniimagenet.sh │ ├── matchingold_rn12_tiered.sh │ ├── nca_rn12_cifar.sh │ ├── nca_rn12_cifar_soft.sh │ ├── nca_rn12_miniimagenet.sh │ ├── nca_rn12_miniimagenet_soft.sh │ ├── nca_rn12_tiered.sh │ ├── nca_rn12_tiered_soft.sh │ ├── protonew_rn12_cifar.sh │ ├── protonew_rn12_tiered.sh │ ├── protoold_rn12_cifar.sh │ ├── protoold_rn12_miniimagenet.sh │ ├── protoold_rn12_tiered.sh │ ├── sample_NCA_pairs_matching.sh │ ├── sample_NCA_pairs_proto.sh │ ├── simpleshot_params.sh │ ├── simpleshot_rn12_cifar.sh │ ├── simpleshot_rn12_miniimagenet.sh │ ├── simpleshot_rn12_tiered.sh │ └── supcont_rn12_miniimagenet.sh ├── cifarfs_split.py ├── extract_mean_conf.py ├── nca_vs_protonets_ablations.py ├── nca_vs_protonets_batchsize.py └── start_training.py ├── setup.py ├── split └── .gitignore └── src ├── __init__.py ├── configs ├── __init__.py ├── configuration.py ├── get_configs.py └── load_yaml.py ├── datasets ├── __init__.py ├── loader.py ├── sampler.py ├── tieredImagenet.py └── transform.py ├── models ├── ResNet.py └── __init__.py ├── train ├── __init__.py ├── loss.py ├── opt_supportset.py └── train.py └── utils ├── __init__.py ├── evaluation.py ├── logs.py └── meters.py /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fiveai/on-episodes-fsl/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fiveai/on-episodes-fsl/HEAD/README.md -------------------------------------------------------------------------------- /dataset_configs/CIFARFS.yaml.example: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fiveai/on-episodes-fsl/HEAD/dataset_configs/CIFARFS.yaml.example -------------------------------------------------------------------------------- /dataset_configs/miniimagenet.yaml.example: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fiveai/on-episodes-fsl/HEAD/dataset_configs/miniimagenet.yaml.example -------------------------------------------------------------------------------- /dataset_configs/tieredimagenet.yaml.example: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fiveai/on-episodes-fsl/HEAD/dataset_configs/tieredimagenet.yaml.example -------------------------------------------------------------------------------- /environment.pytorch13.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fiveai/on-episodes-fsl/HEAD/environment.pytorch13.yml -------------------------------------------------------------------------------- /expm_args/.gitignore: -------------------------------------------------------------------------------- 1 | * 2 | !.gitignore 3 | -------------------------------------------------------------------------------- /results/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fiveai/on-episodes-fsl/HEAD/results/.gitignore -------------------------------------------------------------------------------- /runs/.gitignore: -------------------------------------------------------------------------------- 1 | * 2 | !.gitignore 3 | -------------------------------------------------------------------------------- /scripts/bash_scripts/abl_all_cifar_proto.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fiveai/on-episodes-fsl/HEAD/scripts/bash_scripts/abl_all_cifar_proto.sh -------------------------------------------------------------------------------- /scripts/bash_scripts/abl_all_miniimagenet_proto.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fiveai/on-episodes-fsl/HEAD/scripts/bash_scripts/abl_all_miniimagenet_proto.sh -------------------------------------------------------------------------------- /scripts/bash_scripts/abl_nca_fixed_batch_composition.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fiveai/on-episodes-fsl/HEAD/scripts/bash_scripts/abl_nca_fixed_batch_composition.sh -------------------------------------------------------------------------------- /scripts/bash_scripts/abl_nca_fixed_batch_composition_128.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fiveai/on-episodes-fsl/HEAD/scripts/bash_scripts/abl_nca_fixed_batch_composition_128.sh -------------------------------------------------------------------------------- /scripts/bash_scripts/abl_nca_fixed_batch_composition_512.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fiveai/on-episodes-fsl/HEAD/scripts/bash_scripts/abl_nca_fixed_batch_composition_512.sh -------------------------------------------------------------------------------- /scripts/bash_scripts/abl_nca_repl_sampling.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fiveai/on-episodes-fsl/HEAD/scripts/bash_scripts/abl_nca_repl_sampling.sh -------------------------------------------------------------------------------- /scripts/bash_scripts/abl_nca_repl_sampling_128.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fiveai/on-episodes-fsl/HEAD/scripts/bash_scripts/abl_nca_repl_sampling_128.sh -------------------------------------------------------------------------------- /scripts/bash_scripts/abl_nca_repl_sampling_512.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fiveai/on-episodes-fsl/HEAD/scripts/bash_scripts/abl_nca_repl_sampling_512.sh -------------------------------------------------------------------------------- /scripts/bash_scripts/abl_proto_no_SQ_separation.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fiveai/on-episodes-fsl/HEAD/scripts/bash_scripts/abl_proto_no_SQ_separation.sh -------------------------------------------------------------------------------- /scripts/bash_scripts/abl_proto_no_SQ_separation_128.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fiveai/on-episodes-fsl/HEAD/scripts/bash_scripts/abl_proto_no_SQ_separation_128.sh -------------------------------------------------------------------------------- /scripts/bash_scripts/abl_proto_no_SQ_separation_512.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fiveai/on-episodes-fsl/HEAD/scripts/bash_scripts/abl_proto_no_SQ_separation_512.sh -------------------------------------------------------------------------------- /scripts/bash_scripts/abl_proto_no_aggregates.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fiveai/on-episodes-fsl/HEAD/scripts/bash_scripts/abl_proto_no_aggregates.sh -------------------------------------------------------------------------------- /scripts/bash_scripts/abl_proto_no_aggregates_128.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fiveai/on-episodes-fsl/HEAD/scripts/bash_scripts/abl_proto_no_aggregates_128.sh -------------------------------------------------------------------------------- /scripts/bash_scripts/abl_proto_no_aggregates_512.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fiveai/on-episodes-fsl/HEAD/scripts/bash_scripts/abl_proto_no_aggregates_512.sh -------------------------------------------------------------------------------- /scripts/bash_scripts/batch_expm_cifar_matching.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fiveai/on-episodes-fsl/HEAD/scripts/bash_scripts/batch_expm_cifar_matching.sh -------------------------------------------------------------------------------- /scripts/bash_scripts/batch_expm_cifar_proto.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fiveai/on-episodes-fsl/HEAD/scripts/bash_scripts/batch_expm_cifar_proto.sh -------------------------------------------------------------------------------- /scripts/bash_scripts/batch_expm_miniimagenet_matching.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fiveai/on-episodes-fsl/HEAD/scripts/bash_scripts/batch_expm_miniimagenet_matching.sh -------------------------------------------------------------------------------- /scripts/bash_scripts/batch_expm_miniimagenet_proto.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fiveai/on-episodes-fsl/HEAD/scripts/bash_scripts/batch_expm_miniimagenet_proto.sh -------------------------------------------------------------------------------- /scripts/bash_scripts/generic_nca.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fiveai/on-episodes-fsl/HEAD/scripts/bash_scripts/generic_nca.sh -------------------------------------------------------------------------------- /scripts/bash_scripts/generic_nca_matching.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fiveai/on-episodes-fsl/HEAD/scripts/bash_scripts/generic_nca_matching.sh -------------------------------------------------------------------------------- /scripts/bash_scripts/generic_proto.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fiveai/on-episodes-fsl/HEAD/scripts/bash_scripts/generic_proto.sh -------------------------------------------------------------------------------- /scripts/bash_scripts/generic_proto_matching.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fiveai/on-episodes-fsl/HEAD/scripts/bash_scripts/generic_proto_matching.sh -------------------------------------------------------------------------------- /scripts/bash_scripts/matching_nets_ablation_example.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fiveai/on-episodes-fsl/HEAD/scripts/bash_scripts/matching_nets_ablation_example.sh -------------------------------------------------------------------------------- /scripts/bash_scripts/matchingnew_cifar.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fiveai/on-episodes-fsl/HEAD/scripts/bash_scripts/matchingnew_cifar.sh -------------------------------------------------------------------------------- /scripts/bash_scripts/matchingnew_rn12_tiered.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fiveai/on-episodes-fsl/HEAD/scripts/bash_scripts/matchingnew_rn12_tiered.sh -------------------------------------------------------------------------------- /scripts/bash_scripts/matchingold_rn12_cifar.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fiveai/on-episodes-fsl/HEAD/scripts/bash_scripts/matchingold_rn12_cifar.sh -------------------------------------------------------------------------------- /scripts/bash_scripts/matchingold_rn12_miniimagenet.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fiveai/on-episodes-fsl/HEAD/scripts/bash_scripts/matchingold_rn12_miniimagenet.sh -------------------------------------------------------------------------------- /scripts/bash_scripts/matchingold_rn12_tiered.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fiveai/on-episodes-fsl/HEAD/scripts/bash_scripts/matchingold_rn12_tiered.sh -------------------------------------------------------------------------------- /scripts/bash_scripts/nca_rn12_cifar.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fiveai/on-episodes-fsl/HEAD/scripts/bash_scripts/nca_rn12_cifar.sh -------------------------------------------------------------------------------- /scripts/bash_scripts/nca_rn12_cifar_soft.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fiveai/on-episodes-fsl/HEAD/scripts/bash_scripts/nca_rn12_cifar_soft.sh -------------------------------------------------------------------------------- /scripts/bash_scripts/nca_rn12_miniimagenet.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fiveai/on-episodes-fsl/HEAD/scripts/bash_scripts/nca_rn12_miniimagenet.sh -------------------------------------------------------------------------------- /scripts/bash_scripts/nca_rn12_miniimagenet_soft.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fiveai/on-episodes-fsl/HEAD/scripts/bash_scripts/nca_rn12_miniimagenet_soft.sh -------------------------------------------------------------------------------- /scripts/bash_scripts/nca_rn12_tiered.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fiveai/on-episodes-fsl/HEAD/scripts/bash_scripts/nca_rn12_tiered.sh -------------------------------------------------------------------------------- /scripts/bash_scripts/nca_rn12_tiered_soft.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fiveai/on-episodes-fsl/HEAD/scripts/bash_scripts/nca_rn12_tiered_soft.sh -------------------------------------------------------------------------------- /scripts/bash_scripts/protonew_rn12_cifar.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fiveai/on-episodes-fsl/HEAD/scripts/bash_scripts/protonew_rn12_cifar.sh -------------------------------------------------------------------------------- /scripts/bash_scripts/protonew_rn12_tiered.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fiveai/on-episodes-fsl/HEAD/scripts/bash_scripts/protonew_rn12_tiered.sh -------------------------------------------------------------------------------- /scripts/bash_scripts/protoold_rn12_cifar.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fiveai/on-episodes-fsl/HEAD/scripts/bash_scripts/protoold_rn12_cifar.sh -------------------------------------------------------------------------------- /scripts/bash_scripts/protoold_rn12_miniimagenet.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fiveai/on-episodes-fsl/HEAD/scripts/bash_scripts/protoold_rn12_miniimagenet.sh -------------------------------------------------------------------------------- /scripts/bash_scripts/protoold_rn12_tiered.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fiveai/on-episodes-fsl/HEAD/scripts/bash_scripts/protoold_rn12_tiered.sh -------------------------------------------------------------------------------- /scripts/bash_scripts/sample_NCA_pairs_matching.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fiveai/on-episodes-fsl/HEAD/scripts/bash_scripts/sample_NCA_pairs_matching.sh -------------------------------------------------------------------------------- /scripts/bash_scripts/sample_NCA_pairs_proto.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fiveai/on-episodes-fsl/HEAD/scripts/bash_scripts/sample_NCA_pairs_proto.sh -------------------------------------------------------------------------------- /scripts/bash_scripts/simpleshot_params.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fiveai/on-episodes-fsl/HEAD/scripts/bash_scripts/simpleshot_params.sh -------------------------------------------------------------------------------- /scripts/bash_scripts/simpleshot_rn12_cifar.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fiveai/on-episodes-fsl/HEAD/scripts/bash_scripts/simpleshot_rn12_cifar.sh -------------------------------------------------------------------------------- /scripts/bash_scripts/simpleshot_rn12_miniimagenet.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fiveai/on-episodes-fsl/HEAD/scripts/bash_scripts/simpleshot_rn12_miniimagenet.sh -------------------------------------------------------------------------------- /scripts/bash_scripts/simpleshot_rn12_tiered.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fiveai/on-episodes-fsl/HEAD/scripts/bash_scripts/simpleshot_rn12_tiered.sh -------------------------------------------------------------------------------- /scripts/bash_scripts/supcont_rn12_miniimagenet.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fiveai/on-episodes-fsl/HEAD/scripts/bash_scripts/supcont_rn12_miniimagenet.sh -------------------------------------------------------------------------------- /scripts/cifarfs_split.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fiveai/on-episodes-fsl/HEAD/scripts/cifarfs_split.py -------------------------------------------------------------------------------- /scripts/extract_mean_conf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fiveai/on-episodes-fsl/HEAD/scripts/extract_mean_conf.py -------------------------------------------------------------------------------- /scripts/nca_vs_protonets_ablations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fiveai/on-episodes-fsl/HEAD/scripts/nca_vs_protonets_ablations.py -------------------------------------------------------------------------------- /scripts/nca_vs_protonets_batchsize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fiveai/on-episodes-fsl/HEAD/scripts/nca_vs_protonets_batchsize.py -------------------------------------------------------------------------------- /scripts/start_training.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fiveai/on-episodes-fsl/HEAD/scripts/start_training.py -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fiveai/on-episodes-fsl/HEAD/setup.py -------------------------------------------------------------------------------- /split/.gitignore: -------------------------------------------------------------------------------- 1 | * 2 | !.gitignore 3 | -------------------------------------------------------------------------------- /src/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/configs/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/configs/configuration.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fiveai/on-episodes-fsl/HEAD/src/configs/configuration.py -------------------------------------------------------------------------------- /src/configs/get_configs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fiveai/on-episodes-fsl/HEAD/src/configs/get_configs.py -------------------------------------------------------------------------------- /src/configs/load_yaml.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fiveai/on-episodes-fsl/HEAD/src/configs/load_yaml.py -------------------------------------------------------------------------------- /src/datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fiveai/on-episodes-fsl/HEAD/src/datasets/__init__.py -------------------------------------------------------------------------------- /src/datasets/loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fiveai/on-episodes-fsl/HEAD/src/datasets/loader.py -------------------------------------------------------------------------------- /src/datasets/sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fiveai/on-episodes-fsl/HEAD/src/datasets/sampler.py -------------------------------------------------------------------------------- /src/datasets/tieredImagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fiveai/on-episodes-fsl/HEAD/src/datasets/tieredImagenet.py -------------------------------------------------------------------------------- /src/datasets/transform.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fiveai/on-episodes-fsl/HEAD/src/datasets/transform.py -------------------------------------------------------------------------------- /src/models/ResNet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fiveai/on-episodes-fsl/HEAD/src/models/ResNet.py -------------------------------------------------------------------------------- /src/models/__init__.py: -------------------------------------------------------------------------------- 1 | from .ResNet import * -------------------------------------------------------------------------------- /src/train/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/train/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fiveai/on-episodes-fsl/HEAD/src/train/loss.py -------------------------------------------------------------------------------- /src/train/opt_supportset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fiveai/on-episodes-fsl/HEAD/src/train/opt_supportset.py -------------------------------------------------------------------------------- /src/train/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fiveai/on-episodes-fsl/HEAD/src/train/train.py -------------------------------------------------------------------------------- /src/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/utils/evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fiveai/on-episodes-fsl/HEAD/src/utils/evaluation.py -------------------------------------------------------------------------------- /src/utils/logs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fiveai/on-episodes-fsl/HEAD/src/utils/logs.py -------------------------------------------------------------------------------- /src/utils/meters.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fiveai/on-episodes-fsl/HEAD/src/utils/meters.py --------------------------------------------------------------------------------