training script

[IssamLaradji]

great work! is there a training script?

[sajadn]

Hi Isaam,
The model is trained using maximum likelihood, so the training script is what we have in the density estimation of the readme and I'll state it below again:

python3 density_estimation.py --prior exemplar_prior --dataset {dynamic_mnist|fashion_mnist|omniglot} --model_name {vae|hvae_2level|convhvae_2level} --number_components {25000|11500} --approximate_prior {True|False}

This will train the model and you'll have the final model stored in the current directory.

Sajad