Lower test parameters to run on travis

PiperOrigin-RevId: 219725054
goldenbean · Nov 2, 2018 · 290d90c · 290d90c
1 parent e57f310
commit 290d90c
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/probabilistic_vqvae/run.sh b/probabilistic_vqvae/run.sh
@@ -9,4 +9,6 @@ pip install tf-nightly
 pip install tfp-nightly
 pip install -r probabilistic_vqvae/requirements.txt
 python -m probabilistic_vqvae.mnist_experiments \
-          --max_steps=500 --latent_size 2 --code_size 4 --base_depth 8 --batch_size 16
+          --max_steps=500 \
+          --latent_size 2 --num_codes 2 --code_size 4 --base_depth 2 --batch_size 4 \
+          --nouse_autoregressive_prior