Skip to content
No description, website, or topics provided.
Python C++
Branch: master
Clone or download

Generative Teaching Networks

Official PyTorch implementation of Generative Teaching Networks. For research purpose only. Support and/or new releases may be limited.


Clone the repo:

git clone && 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.

Training on MNIST

To train on MNIST simply run python This command reads experiments/cgtn.json to get arguments.

Architecture Search on CIFAR10

The architecture search experiments is a separate component based on NAO which has a different LICENSE agreement. See architecture_search/ for more details.

You can’t perform that action at this time.