Chainer IMSAT
Implementation of Learning Discrete Representations via Information Maximizing Self-Augmented Training (IMSAT) in Chainer.
Requirements
chainer 3.0.0
cupy 2.0.0
munkres
How to train
python train_cluster.py --epoch 50 --batchsize 250 --gpu 0 --dataset mnist
python train_cluster.py --epoch 50 --batchsize 250 --gpu 0 --dataset 20news
Reference