You can train any VAE baseline model via: python run_experiments_real.py [name] --model cci_vae [arguments] For example, to run CCI VAE and variants on rotated digits: python run_experiments_real.py ...