Skip to content

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

Notifications You must be signed in to change notification settings

TDeVries/enas_pytorch

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

10 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

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.

About

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

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages