/enas_pytorch

PyTorch port of "Efficient Neural Architecture Search via Parameters Sharing"

Primary LanguagePython

Efficient Neural Architecture Search (ENAS)

This repo contains a PyTorch implementation of Efficient Neural Architecture Search via Parameters Sharing.

This implementation is a port of the official Tensorflow implementation. As such, I have tried to replicate hyperparameter settings and "secret sauce" tricks of the original implementation as closely as possible, although there still appears to be some differences in performance.

Currently only the CNN macro architecture search has been implemented. For a PyTorch implementation of RNN cell search see carpedm20's ENAS repo.