/NAO_pytorch

Pytorch Implementation of Neural Architecture Optimization

Primary LanguagePythonGNU General Public License v3.0GPL-3.0

Neural Architecture Optimization

This is the library of Neural Architecture Optimization

The library currently contains following sub-libraries:

  1. NAO-V1: Official pytorch implementation of cnn part of Neural Architecture Optimization, by Renqian Luo*, Fei Tian*, Tao Qin, En-Hong Chen, Tie-Yan Liu. *=equal contribution
  2. NAO-V2: Official implementation of NAO-V2: Balanced One-shot Neural Architecture Optimization, by Renqian Luo, Tao Qin, En-Hong Chen

NAO-V1

Released with the paper Neural Architecture Optimization, by Renqian Luo*, Fei Tian*, Tao Qin, En-Hong Chen, Tie-Yan Liu. *=equal contribution

NAO-V2

Released with the paper Balanced One-shot Neural Architecture Optimization, by Renqian Luo, Tao Qin, En-Hong Chen

License

The codes and models in this repo are released under the GNU GPLv3 license.

This is not an official Microsoft product.