├── .gitignore ├── LICENSE ├── README.md ├── configs ├── ablation │ ├── 1x1conv_init │ │ ├── vgg_nyuv2_init_0.0_1.0.yaml │ │ ├── vgg_nyuv2_init_0.1_0.9.yaml │ │ ├── vgg_nyuv2_init_0.2_0.8.yaml │ │ ├── vgg_nyuv2_init_0.5_0.5.yaml │ │ ├── vgg_nyuv2_init_0.8_0.2.yaml │ │ ├── vgg_nyuv2_init_0.9_0.1.yaml │ │ ├── vgg_nyuv2_init_1.0_0.0.yaml │ │ └── vgg_nyuv2_init_random.yaml │ ├── arch_lr │ │ ├── vgg_nyuv2_1000x_lr.yaml │ │ ├── vgg_nyuv2_100x_lr.yaml │ │ ├── vgg_nyuv2_10x_lr.yaml │ │ └── vgg_nyuv2_1x_lr.yaml │ ├── search_algo │ │ ├── vgg_nyuv2_D_Relax_D_Discre.yaml │ │ ├── vgg_nyuv2_D_Relax_D_Discre_Entropy.yaml │ │ ├── vgg_nyuv2_D_Relax_S_Discre.yaml │ │ ├── vgg_nyuv2_D_Relax_S_Discre_Entropy.yaml │ │ ├── vgg_nyuv2_S_Relax_D_Discre.yaml │ │ ├── vgg_nyuv2_S_Relax_D_Discre_Entropy.yaml │ │ ├── vgg_nyuv2_S_Relax_S_Discre.yaml │ │ └── vgg_nyuv2_S_Relax_S_Discre_Entropy.yaml │ └── vgg_nyuv2_default.yaml └── vgg │ ├── vgg_nyuv2_mtlnas_supernet.yaml │ ├── vgg_nyuv2_nddr.yaml │ └── vgg_nyuv2_nddr_supernet.yaml ├── core ├── __init__.py ├── config │ ├── __init__.py │ └── defaults.py ├── data │ ├── __init__.py │ └── loader.py ├── models │ ├── __init__.py │ ├── common_layers.py │ ├── nddr_net.py │ ├── supernet.py │ ├── vgg16_lfov_bn.py │ └── vgg16_lfov_bn_16_stages.py ├── tasks.py └── utils │ ├── __init__.py │ ├── losses.py │ ├── metrics.py │ └── visualization.py ├── figures └── overview.png ├── requirements.txt └── tools ├── eval.py ├── eval_nas.py ├── train.py └── train_nas.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bhpfelix/MTLNAS/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bhpfelix/MTLNAS/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bhpfelix/MTLNAS/HEAD/README.md -------------------------------------------------------------------------------- /configs/ablation/1x1conv_init/vgg_nyuv2_init_0.0_1.0.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bhpfelix/MTLNAS/HEAD/configs/ablation/1x1conv_init/vgg_nyuv2_init_0.0_1.0.yaml -------------------------------------------------------------------------------- /configs/ablation/1x1conv_init/vgg_nyuv2_init_0.1_0.9.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bhpfelix/MTLNAS/HEAD/configs/ablation/1x1conv_init/vgg_nyuv2_init_0.1_0.9.yaml -------------------------------------------------------------------------------- /configs/ablation/1x1conv_init/vgg_nyuv2_init_0.2_0.8.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bhpfelix/MTLNAS/HEAD/configs/ablation/1x1conv_init/vgg_nyuv2_init_0.2_0.8.yaml -------------------------------------------------------------------------------- /configs/ablation/1x1conv_init/vgg_nyuv2_init_0.5_0.5.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bhpfelix/MTLNAS/HEAD/configs/ablation/1x1conv_init/vgg_nyuv2_init_0.5_0.5.yaml -------------------------------------------------------------------------------- /configs/ablation/1x1conv_init/vgg_nyuv2_init_0.8_0.2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bhpfelix/MTLNAS/HEAD/configs/ablation/1x1conv_init/vgg_nyuv2_init_0.8_0.2.yaml -------------------------------------------------------------------------------- /configs/ablation/1x1conv_init/vgg_nyuv2_init_0.9_0.1.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bhpfelix/MTLNAS/HEAD/configs/ablation/1x1conv_init/vgg_nyuv2_init_0.9_0.1.yaml -------------------------------------------------------------------------------- /configs/ablation/1x1conv_init/vgg_nyuv2_init_1.0_0.0.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bhpfelix/MTLNAS/HEAD/configs/ablation/1x1conv_init/vgg_nyuv2_init_1.0_0.0.yaml -------------------------------------------------------------------------------- /configs/ablation/1x1conv_init/vgg_nyuv2_init_random.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bhpfelix/MTLNAS/HEAD/configs/ablation/1x1conv_init/vgg_nyuv2_init_random.yaml -------------------------------------------------------------------------------- /configs/ablation/arch_lr/vgg_nyuv2_1000x_lr.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bhpfelix/MTLNAS/HEAD/configs/ablation/arch_lr/vgg_nyuv2_1000x_lr.yaml -------------------------------------------------------------------------------- /configs/ablation/arch_lr/vgg_nyuv2_100x_lr.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bhpfelix/MTLNAS/HEAD/configs/ablation/arch_lr/vgg_nyuv2_100x_lr.yaml -------------------------------------------------------------------------------- /configs/ablation/arch_lr/vgg_nyuv2_10x_lr.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bhpfelix/MTLNAS/HEAD/configs/ablation/arch_lr/vgg_nyuv2_10x_lr.yaml -------------------------------------------------------------------------------- /configs/ablation/arch_lr/vgg_nyuv2_1x_lr.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bhpfelix/MTLNAS/HEAD/configs/ablation/arch_lr/vgg_nyuv2_1x_lr.yaml -------------------------------------------------------------------------------- /configs/ablation/search_algo/vgg_nyuv2_D_Relax_D_Discre.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bhpfelix/MTLNAS/HEAD/configs/ablation/search_algo/vgg_nyuv2_D_Relax_D_Discre.yaml -------------------------------------------------------------------------------- /configs/ablation/search_algo/vgg_nyuv2_D_Relax_D_Discre_Entropy.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bhpfelix/MTLNAS/HEAD/configs/ablation/search_algo/vgg_nyuv2_D_Relax_D_Discre_Entropy.yaml -------------------------------------------------------------------------------- /configs/ablation/search_algo/vgg_nyuv2_D_Relax_S_Discre.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bhpfelix/MTLNAS/HEAD/configs/ablation/search_algo/vgg_nyuv2_D_Relax_S_Discre.yaml -------------------------------------------------------------------------------- /configs/ablation/search_algo/vgg_nyuv2_D_Relax_S_Discre_Entropy.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bhpfelix/MTLNAS/HEAD/configs/ablation/search_algo/vgg_nyuv2_D_Relax_S_Discre_Entropy.yaml -------------------------------------------------------------------------------- /configs/ablation/search_algo/vgg_nyuv2_S_Relax_D_Discre.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bhpfelix/MTLNAS/HEAD/configs/ablation/search_algo/vgg_nyuv2_S_Relax_D_Discre.yaml -------------------------------------------------------------------------------- /configs/ablation/search_algo/vgg_nyuv2_S_Relax_D_Discre_Entropy.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bhpfelix/MTLNAS/HEAD/configs/ablation/search_algo/vgg_nyuv2_S_Relax_D_Discre_Entropy.yaml -------------------------------------------------------------------------------- /configs/ablation/search_algo/vgg_nyuv2_S_Relax_S_Discre.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bhpfelix/MTLNAS/HEAD/configs/ablation/search_algo/vgg_nyuv2_S_Relax_S_Discre.yaml -------------------------------------------------------------------------------- /configs/ablation/search_algo/vgg_nyuv2_S_Relax_S_Discre_Entropy.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bhpfelix/MTLNAS/HEAD/configs/ablation/search_algo/vgg_nyuv2_S_Relax_S_Discre_Entropy.yaml -------------------------------------------------------------------------------- /configs/ablation/vgg_nyuv2_default.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bhpfelix/MTLNAS/HEAD/configs/ablation/vgg_nyuv2_default.yaml -------------------------------------------------------------------------------- /configs/vgg/vgg_nyuv2_mtlnas_supernet.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bhpfelix/MTLNAS/HEAD/configs/vgg/vgg_nyuv2_mtlnas_supernet.yaml -------------------------------------------------------------------------------- /configs/vgg/vgg_nyuv2_nddr.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bhpfelix/MTLNAS/HEAD/configs/vgg/vgg_nyuv2_nddr.yaml -------------------------------------------------------------------------------- /configs/vgg/vgg_nyuv2_nddr_supernet.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bhpfelix/MTLNAS/HEAD/configs/vgg/vgg_nyuv2_nddr_supernet.yaml -------------------------------------------------------------------------------- /core/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /core/config/__init__.py: -------------------------------------------------------------------------------- 1 | from .defaults import _C as cfg 2 | -------------------------------------------------------------------------------- /core/config/defaults.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bhpfelix/MTLNAS/HEAD/core/config/defaults.py -------------------------------------------------------------------------------- /core/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bhpfelix/MTLNAS/HEAD/core/data/__init__.py -------------------------------------------------------------------------------- /core/data/loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bhpfelix/MTLNAS/HEAD/core/data/loader.py -------------------------------------------------------------------------------- /core/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bhpfelix/MTLNAS/HEAD/core/models/__init__.py -------------------------------------------------------------------------------- /core/models/common_layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bhpfelix/MTLNAS/HEAD/core/models/common_layers.py -------------------------------------------------------------------------------- /core/models/nddr_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bhpfelix/MTLNAS/HEAD/core/models/nddr_net.py -------------------------------------------------------------------------------- /core/models/supernet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bhpfelix/MTLNAS/HEAD/core/models/supernet.py -------------------------------------------------------------------------------- /core/models/vgg16_lfov_bn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bhpfelix/MTLNAS/HEAD/core/models/vgg16_lfov_bn.py -------------------------------------------------------------------------------- /core/models/vgg16_lfov_bn_16_stages.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bhpfelix/MTLNAS/HEAD/core/models/vgg16_lfov_bn_16_stages.py -------------------------------------------------------------------------------- /core/tasks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bhpfelix/MTLNAS/HEAD/core/tasks.py -------------------------------------------------------------------------------- /core/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bhpfelix/MTLNAS/HEAD/core/utils/__init__.py -------------------------------------------------------------------------------- /core/utils/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bhpfelix/MTLNAS/HEAD/core/utils/losses.py -------------------------------------------------------------------------------- /core/utils/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bhpfelix/MTLNAS/HEAD/core/utils/metrics.py -------------------------------------------------------------------------------- /core/utils/visualization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bhpfelix/MTLNAS/HEAD/core/utils/visualization.py -------------------------------------------------------------------------------- /figures/overview.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bhpfelix/MTLNAS/HEAD/figures/overview.png -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bhpfelix/MTLNAS/HEAD/requirements.txt -------------------------------------------------------------------------------- /tools/eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bhpfelix/MTLNAS/HEAD/tools/eval.py -------------------------------------------------------------------------------- /tools/eval_nas.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bhpfelix/MTLNAS/HEAD/tools/eval_nas.py -------------------------------------------------------------------------------- /tools/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bhpfelix/MTLNAS/HEAD/tools/train.py -------------------------------------------------------------------------------- /tools/train_nas.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bhpfelix/MTLNAS/HEAD/tools/train_nas.py --------------------------------------------------------------------------------