Skip to content

CS394R Final Project : Searching Efficient Neural Architectures using Reinforcement Learning

Notifications You must be signed in to change notification settings

TharunMohandoss/EfficientNeuralArchitectureSearch

Repository files navigation

EfficientNeuralArchitectureSearch

This is an implementation of the NAS framework in pytorch including a complete rewrite of the NAS Layer.

Instructions :

python3 train.py

Libraries Required :

  • torch
  • torchvision
  • thop

Contributors :

Reuse policy :

Do credit this work if you reuse this code!

Results :

The results are detailed in the project report(Find the pdf file in the repository itself).

About

CS394R Final Project : Searching Efficient Neural Architectures using Reinforcement Learning

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages