Pytorch implementation of Deep Variational Information Bottleneck(paper, original code)

python 3.6.4 pytorch 0.3.1.post2 tensorboardX(optional) tensorflow(optional) 

python main.py --beta 1e-3 --num_avg 12 

1. Deep Learning and the Information Bottleneck Principle, Tishby et al.
2. Deep Variational Information Bottleneck, Alemi et al.
3. Tensorflow Demo : https://github.com/alexalemi/vib_demo