├── README.md ├── configs ├── config.py └── subnets │ ├── 314.yaml │ ├── 472.yaml │ └── 574.yaml ├── install.sh ├── lib ├── core │ ├── train_function.py │ └── valid_function.py ├── dataset │ ├── __init__.py │ ├── _init_paths.py │ ├── auto_augment.py │ ├── base_dataset.py │ ├── cifar.py │ ├── cifar_utils.py │ ├── loader.py │ ├── processing.py │ ├── transform.py │ └── utils.py ├── models │ ├── NEAS_builder.py │ ├── NEAS_subnet.py │ ├── units.py │ └── utils.py └── utils │ ├── EMA.py │ ├── __init__.py │ ├── flops_table.py │ ├── helpers.py │ ├── loss.py │ ├── optimizer.py │ ├── saver.py │ ├── scheduler.py │ └── util.py └── tools ├── __init__.py ├── _init_paths.py ├── distributed_test.sh ├── retrain.py └── utils.py /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/researchmm/NEAS/HEAD/README.md -------------------------------------------------------------------------------- /configs/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/researchmm/NEAS/HEAD/configs/config.py -------------------------------------------------------------------------------- /configs/subnets/314.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/researchmm/NEAS/HEAD/configs/subnets/314.yaml -------------------------------------------------------------------------------- /configs/subnets/472.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/researchmm/NEAS/HEAD/configs/subnets/472.yaml -------------------------------------------------------------------------------- /configs/subnets/574.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/researchmm/NEAS/HEAD/configs/subnets/574.yaml -------------------------------------------------------------------------------- /install.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/researchmm/NEAS/HEAD/install.sh -------------------------------------------------------------------------------- /lib/core/train_function.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/researchmm/NEAS/HEAD/lib/core/train_function.py -------------------------------------------------------------------------------- /lib/core/valid_function.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/researchmm/NEAS/HEAD/lib/core/valid_function.py -------------------------------------------------------------------------------- /lib/dataset/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/researchmm/NEAS/HEAD/lib/dataset/__init__.py -------------------------------------------------------------------------------- /lib/dataset/_init_paths.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/researchmm/NEAS/HEAD/lib/dataset/_init_paths.py -------------------------------------------------------------------------------- /lib/dataset/auto_augment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/researchmm/NEAS/HEAD/lib/dataset/auto_augment.py -------------------------------------------------------------------------------- /lib/dataset/base_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/researchmm/NEAS/HEAD/lib/dataset/base_dataset.py -------------------------------------------------------------------------------- /lib/dataset/cifar.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/researchmm/NEAS/HEAD/lib/dataset/cifar.py -------------------------------------------------------------------------------- /lib/dataset/cifar_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/researchmm/NEAS/HEAD/lib/dataset/cifar_utils.py -------------------------------------------------------------------------------- /lib/dataset/loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/researchmm/NEAS/HEAD/lib/dataset/loader.py -------------------------------------------------------------------------------- /lib/dataset/processing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/researchmm/NEAS/HEAD/lib/dataset/processing.py -------------------------------------------------------------------------------- /lib/dataset/transform.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/researchmm/NEAS/HEAD/lib/dataset/transform.py -------------------------------------------------------------------------------- /lib/dataset/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/researchmm/NEAS/HEAD/lib/dataset/utils.py -------------------------------------------------------------------------------- /lib/models/NEAS_builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/researchmm/NEAS/HEAD/lib/models/NEAS_builder.py -------------------------------------------------------------------------------- /lib/models/NEAS_subnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/researchmm/NEAS/HEAD/lib/models/NEAS_subnet.py -------------------------------------------------------------------------------- /lib/models/units.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/researchmm/NEAS/HEAD/lib/models/units.py -------------------------------------------------------------------------------- /lib/models/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/researchmm/NEAS/HEAD/lib/models/utils.py -------------------------------------------------------------------------------- /lib/utils/EMA.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/researchmm/NEAS/HEAD/lib/utils/EMA.py -------------------------------------------------------------------------------- /lib/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /lib/utils/flops_table.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/researchmm/NEAS/HEAD/lib/utils/flops_table.py -------------------------------------------------------------------------------- /lib/utils/helpers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/researchmm/NEAS/HEAD/lib/utils/helpers.py -------------------------------------------------------------------------------- /lib/utils/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/researchmm/NEAS/HEAD/lib/utils/loss.py -------------------------------------------------------------------------------- /lib/utils/optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/researchmm/NEAS/HEAD/lib/utils/optimizer.py -------------------------------------------------------------------------------- /lib/utils/saver.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/researchmm/NEAS/HEAD/lib/utils/saver.py -------------------------------------------------------------------------------- /lib/utils/scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/researchmm/NEAS/HEAD/lib/utils/scheduler.py -------------------------------------------------------------------------------- /lib/utils/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/researchmm/NEAS/HEAD/lib/utils/util.py -------------------------------------------------------------------------------- /tools/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tools/_init_paths.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/researchmm/NEAS/HEAD/tools/_init_paths.py -------------------------------------------------------------------------------- /tools/distributed_test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/researchmm/NEAS/HEAD/tools/distributed_test.sh -------------------------------------------------------------------------------- /tools/retrain.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/researchmm/NEAS/HEAD/tools/retrain.py -------------------------------------------------------------------------------- /tools/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/researchmm/NEAS/HEAD/tools/utils.py --------------------------------------------------------------------------------