parameter setting

acetylSv · Mar 6, 2018 · 6bcfe06 · 6bcfe06
1 parent c23672e
commit 6bcfe06
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 3 deletions.
diff --git a/.gitignore b/.gitignore
@@ -3,3 +3,4 @@ feat/
 cycle_gan_vc_log/
 get_train_infer.py
 test.py
+get.sh
diff --git a/cycle_gan_graph.py b/cycle_gan_graph.py
@@ -157,8 +157,8 @@ def __init__(self, mode="train"):
 
                 # Training Scheme
                 self.global_step = tf.Variable(0, name='global_step', trainable=False)
-                self.gen_optimizer = tf.train.AdamOptimizer(learning_rate=hp.lr*2)
-                self.dis_optimizer = tf.train.AdamOptimizer(learning_rate=hp.lr)
+                self.gen_optimizer = tf.train.AdamOptimizer(learning_rate=hp.lr*2, beta1=0.5)
+                self.dis_optimizer = tf.train.AdamOptimizer(learning_rate=hp.lr, beta1=0.5)
 
                 with tf.variable_scope('gen_train'):
                     gvs = self.gen_optimizer.compute_gradients(

diff --git a/hyperparams.py b/hyperparams.py
@@ -11,7 +11,7 @@ class Hyperparams:
     lr = 0.0001
     batch_size = 1
     fix_seq_length = 128
-    summary_period = 3
+    summary_period = 300
     save_period = 500
     LAMBDA_CYCLE = 10
     LAMBDA_IDENTITY = 5