Skip to content
/ frog Public

Differentiable neural architecture search

Notifications You must be signed in to change notification settings

bkj/frog

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

52 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

frog

Port of DARTS neural architecture search.

Uses basenet as a lightweight pytorch wrapper.

Models

frog/models/CIFAR

Replication of the CIFAR10 experiment from original DARTS implementation.

The search phase produces exactly the same results as the original implementation.

The final training phase has not been fully implemented yet -- the original implementation has some bells and whistles that are probably necessary to get (near) SOTA performance. There are also some minor intentional difference from DARTS train.py (for simplicity purposes). See this issue for some details.

frog/models/fashion_mnist

Tiny example that shows DARTS outperforming random search. Should run in just a few minutes on a GPU.

Disclaimer

Code under active development -- open an issue and I'll help as best I can

Releases

No releases published

Packages

No packages published