search

sajadn / Exemplar-VAE

Exemplar VAE: Linking Generative Models, Nearest Neighbor Retrieval, and Data Augmentation
https://arxiv.org/abs/2004.04795
MIT License
69 stars 8 forks source link

training script #2

Open IssamLaradji opened 3 years ago

IssamLaradji commented 3 years ago

great work! is there a training script?

sajadn commented 3 years ago

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