enas_pytorch icon indicating copy to clipboard operation
enas_pytorch copied to clipboard

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

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.