Official Torch Code for the paper Guiding InfoGAN with Semi-Supervision.
The script implements the training routine for the MNIST dataset. We used a cropped version of size 28x28 which can be found here.
After cloning this repository, download the dataset into the code directory and specify the path to it in the configuration.lua file, alongside any other parameter changes you wish to have. Then execute the training procedure in train_ss-infogan.lua with
If the score with respect to a classifier is required, a valid classifier must be trained. Its path can then be specified in the configuration.lua file. If none is available, the metric will not be calculated.
luarocks install torchnet luarocks install penlight luarocks install nninit luarocks install cutorch luarocks install cunn
This code is a heavily modified version of and based on the code found here.