automl / automl/NASLib

How to parse any arch in NAS-Bench-101 as pytorch class to train.

Open
#121 0 comments 0 reactions 2 assignees Claimed by @shakibamrd View on GitHub
Dominant language
Python
Stars
595
Forks
133
PR merge metrics
No merged PRs in 30d

Description

Hi,
I'm trying to sample few architectures from NAS-Bench-101, and then query the socre, and also get the torch-model according to the architecutre.

I wanna use the model to do some customer training or testing.

In core/graph.py, it shows we should us the following code to parse the model to pytorch module

```
**Use as pytorch module**
If you want to learn the weights of the operations or any
other parameters of the graph you have to parse it first.
>>> graph = getFancySearchSpace()
>>> graph.parse()
>>> logits = graph(data)
>>> optimizer.min(loss(logits, target))
```

But the graph.parse() will fail if i use NasBench101SearchSpace as graph instance ,

Contributor guide

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.