How to generate a computational graph from an architecture from the size search space?

[ifed-ucsd]

Describe the Question
A clear and concise description of the question.

• Is it about the topology search space in NATS-Bench? No
• Is it about the size search space in NATS-Bench? Yes
• Which figure or table are you referring to in the paper? N/A

I would like to generate a computational graph from a sample from the size search space. Do you provide any functionality for that? I presume that you must have used this functionality when creating this dataset, so if it is not available within the NATS-Bench codebase, can you give some pointers as to how I can do this?

Thanks!

[ifed-ucsd]

After looking at the readme further, I guess you already have this in api.get_net_config?

[D-X-Y]

Yes, please have a look at this notebook: https://github.com/D-X-Y/NATS-Bench/blob/main/notebooks/issue-21.ipynb

In short, config = api.get_net_config(architecture_index, 'cifar10') will give what you want. In this notebook, it shows the details in the config.

[ifed-ucsd]

Thanks for such a fast reply! That solves my issue.

[D-X-Y]

You are always welcome~