← all repositories

carpedm20/ENAS-pytorch

A PyTorch implementation of ENAS that reduces NAS computational cost by 1000x through parameter sharing to discover optimal RNN and CNN architectures.

2.7k stars Python ML Frameworks
ENAS-pytorch
Velocity · 7d
+0.9
★ / day
Trend
steady
star history

This repository implements Efficient Neural Architecture Search (ENAS), an AutoML technique that uses a reinforcement learning controller to search for optimal neural network architectures. The controller learns to sample architectures from a shared computation graph, dramatically reducing GPU-hours compared to traditional NAS. It demonstrates state-of-the-art performance on Penn Treebank language modeling by discovering optimal recurrent cell structures.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.