diff --git a/.gitignore b/.gitignore index d8113ef..ad90e5e 100644 --- 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 index 30e815d..56c32c9 100644 --- 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 index 13a2718..aee0f5b 100644 --- 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