Complexity Aware Supernet Training for One-Shot NAS

Implementation of Complexity Aware Supernet Training for One-Shot NAS.

Pre-requisites

This repository has been tested with the following libraries:

• Python (3.8.8)
• Pytorch (1.8.1)

Preparation and Download

1. Download NAS-Bench-201 API: NAS-Bench-201-v1_1-096897.pth (4.7G) and save it under ./Complexity_Aware_Supernet_Training/data/ folder.

2. Install NasBench201 via pip.

pip install nas-bench-201

Please refer to AutoDL-Projects for more details.

Acknowledgements

Our codes are partly based on the following repositories.

• AutoDL
• GM-NAS