InfoMax-VAE pytorch implementation
Learning Representaion by Maximizaing Mutual Information in Variational Autoencoders in PyTorch
pip install -r requirements.txt
python main.py \--method infomax \--model mlp \--dim 128 \--num_iter 500 \--batch_size 64 \--gamma 20 \--alpha 1 \--dataset cifar10 \--trial 1
Learning Representations by Maximizing Mutual Information in Variational Autoencoders
https://arxiv.org/abs/1912.13361
