normalizing_flow_particle_flow

This code is used for comparing the convergence rate of particle flow with normalizing flow, we evaluate their performance on MNIST dataset.

Running the experiments

To train the vae with particle flow or normalizing flow on MNIST dataset, run the following commands.

Trainning VAE with particle flow

python train_particle_flow.py

Training VAE with normalizing flow

python main_train.py

See the training file for particle flow and the training flie for normalizing flow for more configurations.