cvlab-yonsei / AZ-NAS

An official implementation of "AZ-NAS: Assembling Zero-Cost Proxies for Network Architecture Search" (CVPR 2024) in PyTorch.
GNU General Public License v3.0
43 stars 19 forks source link

Pytorch implementation of AZ-NAS

This repository contains the official implementation of the paper "AZ-NAS: Assembling Zero-Cost Proxies for Network Architecture Search" presented at CVPR 2024.

For detailed information, please visit the project website or read the paper on arXiv.

Requirements

To reproduce our results, we provide Dockerfile to build a Docker image.

Refer to the Dockerfile for details about the environment setup.

Code

Instructions and example code for each search space can be found in the README file within the corresponding folder.

Acknowledgement

Our implementation is based on several open-source projects.

We express our gratitude to the authors and contributors of these projects.

If you use any of their assets, please cite the corresponding papers appropriately:

Citation

If you find our work useful in your research, please cite our paper:

@inproceedings{lee2024assembling,
  title={{AZ-NAS}: Assembling Zero-Cost Proxies for Network Architecture Search},
  author={Lee, Junghyup and Ham, Bumsub},
  booktitle={Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition},
  year={2024}
}