Skip to content

Re-implementation of Neural Architecture Search using Reinforcement Learning

Notifications You must be signed in to change notification settings

ydzhang12345/Neural-Architecture-Search

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

19 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Neural-Architecture-Search

This is a re-implementation of the first neural architecture search paper: https://arxiv.org/abs/1611.01578.

We test our implementation on three setting:

  1. customized toy environment, where the dataset is a multi-modal Gaussian;
  2. A 3-layer fully-connected network on Google speech commands dataset;
  3. A simple CNN on Google speech commands dataset.

To run the code, simply go to the toy-model folder and run

python pg.py

Instructions on how to run the google speech commands dataset will be updated soon. However, you have all the scripts needed in this repo. Please take a look on our report: cs234-final-report.pdf in this repo for more details.

About

Re-implementation of Neural Architecture Search using Reinforcement Learning

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages