initial commit

Naresh1318 · Dec 17, 2017 · c4c247e · c4c247e
1 parent 3380b25
commit c4c247e
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/generate_model.py b/generate_model.py
@@ -131,7 +131,7 @@ def training_data(self):
         train_action = []
         train_target = []
 
-        episode_dir = sorted([self.data_dir + "/train/" + p for p in os.listdir(self.data_dir + "/train/")])[:5]
+        episode_dir = sorted([self.data_dir + "/train/" + p for p in os.listdir(self.data_dir + "/train/")])
         n_episodes = len(episode_dir)
         print("Reading training images!")
         for e_i, episode in enumerate(episode_dir):

diff --git a/generate_model_skip.py b/generate_model_skip.py
@@ -164,9 +164,9 @@ def train(self):
         g_vars = [var for var in t_vars if 'g_' in var.name]
 
         g_learning_rate = tf.train.exponential_decay(self.generator_lr, self.global_step,
-                                                     1, 0.999, staircase=True)
+                                                     1, 0.99999, staircase=True)
         d_learning_rate = tf.train.exponential_decay(self.discriminator_lr, self.global_step,
-                                                     1, 0.999, staircase=True)
+                                                     1, 0.99999, staircase=True)
 
         generator_optimizer = tf.train.AdamOptimizer(g_learning_rate, beta1=self.beta1).minimize(generator_loss,
                                                                                                  var_list=g_vars)