Official PyTorch implementation of Generative Teaching Networks. For research purpose only. Support and/or new releases may be limited.
Clone the repo:
git clone https://github.com/uber-research/GTN.git && cd GTN
We use Python 3.6.2. Requirements can be installed by running:
pip install -r requirements.txt
There is a problem with torch>1.2 so we have a hard requirement on torch==1.2.
After installing the dependencies you can run pip install -e . to compile a custom torch kernel.
To train on MNIST simply run python train_cgtn.py. This command reads experiments/cgtn.json to get arguments.
The architecture search experiments is a separate component based on NAO which has a different LICENSE agreement. See architecture_search/ for more details.