Download datasets and benchmark from NATS-Bench :

https://github.com/D-X-Y/NATS-Bench

Place data files in their respective folders in ./datasets.

Optionally, place the benchmark file NATS-tss-v1_0-3ffb9-simple in ./nas_bench/NATSBenchTSS.

Follow arg options in train_supernet.py to train a NAS-Bench-201 supernet.
Follow arg options in split.py to run supernet splitting.