This is an implementation of the NAS framework in pytorch including a complete rewrite of the NAS Layer.
python3 train.py
- torch
- torchvision
- thop
- Tharun Mohandoss
- Surya S Dwivedi ( https://github.com/surya2191997 )
Do credit this work if you reuse this code!
The results are detailed in the project report(Find the pdf file in the repository itself).