From 84272ee093ee31b78378530a2fb9abf6178c877c Mon Sep 17 00:00:00 2001
From: 1Konny <lee.wonkwang94@gmail.com>
Date: Tue, 17 Apr 2018 10:44:08 +0900
Subject: [PATCH] chores
---
main.py | 6 +++---
run_3dchairs.sh | 2 +-
run_celeba.sh | 2 +-
3 files changed, 5 insertions(+), 5 deletions(-)
diff --git a/main.py b/main.py
index aa0e85d..5e05571 100644
--- a/main.py
+++ b/main.py
@@ -26,18 +26,18 @@ def main(args):
if args.train:
net.train()
else:
- net.show_factorization()
+ net.traverse()
if __name__ == "__main__":
parser = argparse.ArgumentParser(description='toy Beta-VAE')
- parser.add_argument('--train', default=True, type=str2bool, help='train or show_factorization')
+ 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('--batch_size', default=64, type=int, help='batch size')
- parser.add_argument('--z_dim', default=10, type=int, help='dimension of the representation z')
+ parser.add_argument('--z_dim', default=32, type=int, help='dimension of the representation z')
parser.add_argument('--beta', default=6.4, type=float, help='beta parameter for KL-term')
parser.add_argument('--lr', default=1e-4, type=float, help='learning rate')
parser.add_argument('--beta1', default=0.5, type=float, help='Adam optimizer beta1')
diff --git a/run_3dchairs.sh b/run_3dchairs.sh
index b56525e..b22e5b4 100644
--- a/run_3dchairs.sh
+++ b/run_3dchairs.sh
@@ -1,3 +1,3 @@
-#1 /bin/bash
+#! /bin/bash
python main.py --dataset 3dchairs --beta 5 --lr 1e-4 --z_dim 32 --viz_name beta_vae_3dchairs --viz_port 55558
diff --git a/run_celeba.sh b/run_celeba.sh
index 235926e..580786d 100644
--- a/run_celeba.sh
+++ b/run_celeba.sh
@@ -1,3 +1,3 @@
-#1 /bin/bash
+#! /bin/bash
python main.py --dataset celeba --beta 250 --lr 1e-4 --z_dim 32 --viz_name beta_vae_celeba --viz_port 55558
--
GitLab