├── .gitignore ├── DeepAA_evaluate ├── README.md ├── __init__.py ├── augmentations.py ├── autoaugment.py ├── common.py ├── data.py ├── deep_autoaugment.py ├── fast_autoaugment.py ├── imagenet.py ├── lr_scheduler.py ├── metrics.py ├── networks │ ├── __init__.py │ ├── convnet.py │ ├── mlp.py │ ├── resnet.py │ ├── shakeshake │ │ ├── __init__.py │ │ ├── shake_resnet.py │ │ ├── shake_resnext.py │ │ └── shakeshake.py │ └── wideresnet.py ├── train.py └── utils.py ├── DeepAA_search.py ├── DeepAA_utils.py ├── README.md ├── __init__.py ├── aug_lib.py ├── augmentation.py ├── confs ├── resnet50_imagenet_DeepAA_8x256_1.yaml ├── resnet50_imagenet_DeepAA_8x256_2.yaml ├── wresnet28x10_cifar100_DeepAA_1.yaml ├── wresnet28x10_cifar100_DeepAA_1_wd1e-3.yaml ├── wresnet28x10_cifar100_DeepAA_2.yaml ├── wresnet28x10_cifar100_DeepAA_2_wd1e-3.yaml ├── wresnet28x10_cifar100_DeepAA_BatchAug8x_1.yaml ├── wresnet28x10_cifar100_DeepAA_BatchAug8x_2.yaml ├── wresnet28x10_cifar10_DeepAA_1.yaml ├── wresnet28x10_cifar10_DeepAA_1_wd1e-3.yaml ├── wresnet28x10_cifar10_DeepAA_2.yaml ├── wresnet28x10_cifar10_DeepAA_2_wd1e-3.yaml ├── wresnet28x10_cifar10_DeepAA_BatchAug8x_1.yaml └── wresnet28x10_cifar10_DeepAA_BatchAug8x_2.yaml ├── data_generator.py ├── imagenet_data_utils.py ├── images ├── DeepAA.png ├── DeepAA_slideslive.png ├── magnitude_distribution_cifar.png ├── magnitude_distribution_imagenet.png └── operation_distribution.png ├── lr_scheduler.py ├── policy.py ├── policy_port ├── policy_DeepAA_cifar_1.npz ├── policy_DeepAA_cifar_2.npz ├── policy_DeepAA_imagenet_1.npz └── policy_DeepAA_imagenet_2.npz ├── requirements.txt ├── resnet.py ├── resnet_imagenet.py └── utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AIoT-MLSys-Lab/DeepAA/HEAD/.gitignore -------------------------------------------------------------------------------- /DeepAA_evaluate/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AIoT-MLSys-Lab/DeepAA/HEAD/DeepAA_evaluate/README.md -------------------------------------------------------------------------------- /DeepAA_evaluate/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /DeepAA_evaluate/augmentations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AIoT-MLSys-Lab/DeepAA/HEAD/DeepAA_evaluate/augmentations.py -------------------------------------------------------------------------------- /DeepAA_evaluate/autoaugment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AIoT-MLSys-Lab/DeepAA/HEAD/DeepAA_evaluate/autoaugment.py -------------------------------------------------------------------------------- /DeepAA_evaluate/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AIoT-MLSys-Lab/DeepAA/HEAD/DeepAA_evaluate/common.py -------------------------------------------------------------------------------- /DeepAA_evaluate/data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AIoT-MLSys-Lab/DeepAA/HEAD/DeepAA_evaluate/data.py -------------------------------------------------------------------------------- /DeepAA_evaluate/deep_autoaugment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AIoT-MLSys-Lab/DeepAA/HEAD/DeepAA_evaluate/deep_autoaugment.py -------------------------------------------------------------------------------- /DeepAA_evaluate/fast_autoaugment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AIoT-MLSys-Lab/DeepAA/HEAD/DeepAA_evaluate/fast_autoaugment.py -------------------------------------------------------------------------------- /DeepAA_evaluate/imagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AIoT-MLSys-Lab/DeepAA/HEAD/DeepAA_evaluate/imagenet.py -------------------------------------------------------------------------------- /DeepAA_evaluate/lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AIoT-MLSys-Lab/DeepAA/HEAD/DeepAA_evaluate/lr_scheduler.py -------------------------------------------------------------------------------- /DeepAA_evaluate/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AIoT-MLSys-Lab/DeepAA/HEAD/DeepAA_evaluate/metrics.py -------------------------------------------------------------------------------- /DeepAA_evaluate/networks/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AIoT-MLSys-Lab/DeepAA/HEAD/DeepAA_evaluate/networks/__init__.py -------------------------------------------------------------------------------- /DeepAA_evaluate/networks/convnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AIoT-MLSys-Lab/DeepAA/HEAD/DeepAA_evaluate/networks/convnet.py -------------------------------------------------------------------------------- /DeepAA_evaluate/networks/mlp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AIoT-MLSys-Lab/DeepAA/HEAD/DeepAA_evaluate/networks/mlp.py -------------------------------------------------------------------------------- /DeepAA_evaluate/networks/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AIoT-MLSys-Lab/DeepAA/HEAD/DeepAA_evaluate/networks/resnet.py -------------------------------------------------------------------------------- /DeepAA_evaluate/networks/shakeshake/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /DeepAA_evaluate/networks/shakeshake/shake_resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AIoT-MLSys-Lab/DeepAA/HEAD/DeepAA_evaluate/networks/shakeshake/shake_resnet.py -------------------------------------------------------------------------------- /DeepAA_evaluate/networks/shakeshake/shake_resnext.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AIoT-MLSys-Lab/DeepAA/HEAD/DeepAA_evaluate/networks/shakeshake/shake_resnext.py -------------------------------------------------------------------------------- /DeepAA_evaluate/networks/shakeshake/shakeshake.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AIoT-MLSys-Lab/DeepAA/HEAD/DeepAA_evaluate/networks/shakeshake/shakeshake.py -------------------------------------------------------------------------------- /DeepAA_evaluate/networks/wideresnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AIoT-MLSys-Lab/DeepAA/HEAD/DeepAA_evaluate/networks/wideresnet.py -------------------------------------------------------------------------------- /DeepAA_evaluate/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AIoT-MLSys-Lab/DeepAA/HEAD/DeepAA_evaluate/train.py -------------------------------------------------------------------------------- /DeepAA_evaluate/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AIoT-MLSys-Lab/DeepAA/HEAD/DeepAA_evaluate/utils.py -------------------------------------------------------------------------------- /DeepAA_search.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AIoT-MLSys-Lab/DeepAA/HEAD/DeepAA_search.py -------------------------------------------------------------------------------- /DeepAA_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AIoT-MLSys-Lab/DeepAA/HEAD/DeepAA_utils.py -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AIoT-MLSys-Lab/DeepAA/HEAD/README.md -------------------------------------------------------------------------------- /__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /aug_lib.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AIoT-MLSys-Lab/DeepAA/HEAD/aug_lib.py -------------------------------------------------------------------------------- /augmentation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AIoT-MLSys-Lab/DeepAA/HEAD/augmentation.py -------------------------------------------------------------------------------- /confs/resnet50_imagenet_DeepAA_8x256_1.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AIoT-MLSys-Lab/DeepAA/HEAD/confs/resnet50_imagenet_DeepAA_8x256_1.yaml -------------------------------------------------------------------------------- /confs/resnet50_imagenet_DeepAA_8x256_2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AIoT-MLSys-Lab/DeepAA/HEAD/confs/resnet50_imagenet_DeepAA_8x256_2.yaml -------------------------------------------------------------------------------- /confs/wresnet28x10_cifar100_DeepAA_1.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AIoT-MLSys-Lab/DeepAA/HEAD/confs/wresnet28x10_cifar100_DeepAA_1.yaml -------------------------------------------------------------------------------- /confs/wresnet28x10_cifar100_DeepAA_1_wd1e-3.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AIoT-MLSys-Lab/DeepAA/HEAD/confs/wresnet28x10_cifar100_DeepAA_1_wd1e-3.yaml -------------------------------------------------------------------------------- /confs/wresnet28x10_cifar100_DeepAA_2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AIoT-MLSys-Lab/DeepAA/HEAD/confs/wresnet28x10_cifar100_DeepAA_2.yaml -------------------------------------------------------------------------------- /confs/wresnet28x10_cifar100_DeepAA_2_wd1e-3.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AIoT-MLSys-Lab/DeepAA/HEAD/confs/wresnet28x10_cifar100_DeepAA_2_wd1e-3.yaml -------------------------------------------------------------------------------- /confs/wresnet28x10_cifar100_DeepAA_BatchAug8x_1.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AIoT-MLSys-Lab/DeepAA/HEAD/confs/wresnet28x10_cifar100_DeepAA_BatchAug8x_1.yaml -------------------------------------------------------------------------------- /confs/wresnet28x10_cifar100_DeepAA_BatchAug8x_2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AIoT-MLSys-Lab/DeepAA/HEAD/confs/wresnet28x10_cifar100_DeepAA_BatchAug8x_2.yaml -------------------------------------------------------------------------------- /confs/wresnet28x10_cifar10_DeepAA_1.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AIoT-MLSys-Lab/DeepAA/HEAD/confs/wresnet28x10_cifar10_DeepAA_1.yaml -------------------------------------------------------------------------------- /confs/wresnet28x10_cifar10_DeepAA_1_wd1e-3.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AIoT-MLSys-Lab/DeepAA/HEAD/confs/wresnet28x10_cifar10_DeepAA_1_wd1e-3.yaml -------------------------------------------------------------------------------- /confs/wresnet28x10_cifar10_DeepAA_2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AIoT-MLSys-Lab/DeepAA/HEAD/confs/wresnet28x10_cifar10_DeepAA_2.yaml -------------------------------------------------------------------------------- /confs/wresnet28x10_cifar10_DeepAA_2_wd1e-3.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AIoT-MLSys-Lab/DeepAA/HEAD/confs/wresnet28x10_cifar10_DeepAA_2_wd1e-3.yaml -------------------------------------------------------------------------------- /confs/wresnet28x10_cifar10_DeepAA_BatchAug8x_1.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AIoT-MLSys-Lab/DeepAA/HEAD/confs/wresnet28x10_cifar10_DeepAA_BatchAug8x_1.yaml -------------------------------------------------------------------------------- /confs/wresnet28x10_cifar10_DeepAA_BatchAug8x_2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AIoT-MLSys-Lab/DeepAA/HEAD/confs/wresnet28x10_cifar10_DeepAA_BatchAug8x_2.yaml -------------------------------------------------------------------------------- /data_generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AIoT-MLSys-Lab/DeepAA/HEAD/data_generator.py -------------------------------------------------------------------------------- /imagenet_data_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AIoT-MLSys-Lab/DeepAA/HEAD/imagenet_data_utils.py -------------------------------------------------------------------------------- /images/DeepAA.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AIoT-MLSys-Lab/DeepAA/HEAD/images/DeepAA.png -------------------------------------------------------------------------------- /images/DeepAA_slideslive.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AIoT-MLSys-Lab/DeepAA/HEAD/images/DeepAA_slideslive.png -------------------------------------------------------------------------------- /images/magnitude_distribution_cifar.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AIoT-MLSys-Lab/DeepAA/HEAD/images/magnitude_distribution_cifar.png -------------------------------------------------------------------------------- /images/magnitude_distribution_imagenet.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AIoT-MLSys-Lab/DeepAA/HEAD/images/magnitude_distribution_imagenet.png -------------------------------------------------------------------------------- /images/operation_distribution.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AIoT-MLSys-Lab/DeepAA/HEAD/images/operation_distribution.png -------------------------------------------------------------------------------- /lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AIoT-MLSys-Lab/DeepAA/HEAD/lr_scheduler.py -------------------------------------------------------------------------------- /policy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AIoT-MLSys-Lab/DeepAA/HEAD/policy.py -------------------------------------------------------------------------------- /policy_port/policy_DeepAA_cifar_1.npz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AIoT-MLSys-Lab/DeepAA/HEAD/policy_port/policy_DeepAA_cifar_1.npz -------------------------------------------------------------------------------- /policy_port/policy_DeepAA_cifar_2.npz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AIoT-MLSys-Lab/DeepAA/HEAD/policy_port/policy_DeepAA_cifar_2.npz -------------------------------------------------------------------------------- /policy_port/policy_DeepAA_imagenet_1.npz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AIoT-MLSys-Lab/DeepAA/HEAD/policy_port/policy_DeepAA_imagenet_1.npz -------------------------------------------------------------------------------- /policy_port/policy_DeepAA_imagenet_2.npz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AIoT-MLSys-Lab/DeepAA/HEAD/policy_port/policy_DeepAA_imagenet_2.npz -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AIoT-MLSys-Lab/DeepAA/HEAD/requirements.txt -------------------------------------------------------------------------------- /resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AIoT-MLSys-Lab/DeepAA/HEAD/resnet.py -------------------------------------------------------------------------------- /resnet_imagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AIoT-MLSys-Lab/DeepAA/HEAD/resnet_imagenet.py -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AIoT-MLSys-Lab/DeepAA/HEAD/utils.py --------------------------------------------------------------------------------