AIoT-MLSys-Lab / arch2vec

[NeurIPS 2020] "Does Unsupervised Architecture Representation Learning Help Neural Architecture Search?" by Shen Yan, Yu Zheng, Wei Ao, Xiao Zeng, Mi Zhang
Apache License 2.0
49 stars 10 forks source link

model problem #2

Closed outofstyle closed 1 year ago

outofstyle commented 2 years ago

I understand that each architecture is encoded as a different graph and the graph of the architecture representation is embedded. Is this true?

shenyann commented 2 years ago

Yes, https://github.com/MSU-MLSys-Lab/arch2vec/blob/main/models/model.py#L10-L45