.

main.py

@@ -34,7 +34,7 @@ if __name__ == "__main__":
 
     parser.add_argument('--train', default=True, type=str2bool, help='train or traverse')
     parser.add_argument('--cuda', default=True, type=str2bool, help='enable cuda')
-    parser.add_argument('--max_iter', default=1e6, type=int, help='maximum training iteration')
+    parser.add_argument('--max_iter', default=1e6, type=float, help='maximum training iteration')
     parser.add_argument('--batch_size', default=64, type=int, help='batch size')
 
     parser.add_argument('--z_dim', default=32, type=int, help='dimension of the representation z')

run_3dchairs.sh

-#! /bin/bash
+#! /bin/sh
 
-python main.py --dataset 3dchairs --beta 4 --lr 1e-4 --z_dim 10 --viz_name beta_vae_3dchairs --viz_port 55558
+python main.py --dataset 3dchairs --max_iter 1e6 --beta 4 --batch_size 64 --lr 1e-4 --z_dim 10 --viz_name beta_vae_3dchairs --viz_port 55558

run_celeba.sh

-#! /bin/bash
+#! /bin/sh
 
-python main.py --dataset celeba --beta 64 --lr 1e-4 --z_dim 10 --viz_name beta_vae_celeba --viz_port 55558
+python main.py --dataset celeba --max_iter 1e6 --beta 64 --batch_size 64 --lr 1e-4 --z_dim 10 --viz_name beta_vae_celeba --viz_port 55558