## Dependencies
- Python 3.10
- Pytorch 1.13.1
- Pytorch Geometric 2.2.0

Our model is implemented based on [GraphGPS](https://arxiv.org/abs/2205.12454).


## Dataset
In the ranking experiments, we used 5 datasets: [NAS-Bench-101](https://arxiv.org/abs/1902.09635), [NAS-Bench-201](https://arxiv.org/abs/2001.00326), [NAS-Bench-301](https://arxiv.org/abs/2008.09777), [NAS-Bench-ASR](https://openreview.net/forum?id=CU0APx9LMaL), [NAS-Bench-Graph](https://arxiv.org/abs/2206.09166).
In the search experiments, we used [DARTS](https://arxiv.org/abs/1806.09055).
## To run experiments
Run `python train_pp.py` for ranking experiments. You can modify the `config_path` in `train_pp.py` to choose the dataset to use.
Run `python darts/train.py` and `python darts/train_imagenet.py` to retrain the searched architecture on CIFAR-10 and ImageNet.