change sample size

kozistr · Jan 6, 2018 · c3e3669 · c3e3669
1 parent a2ad5a4
commit c3e3669
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/BEGAN/began_model.py b/BEGAN/began_model.py
@@ -157,7 +157,7 @@ def decoder(self, x, reuse=None):
         with tf.variable_scope('decoder', reuse=reuse):
             repeat = int(np.log2(self.input_height)) - 2
 
-            x = tf.layers.dense(x, units=self.z_dim * 8 * 8, name='dec-fc-1')
+            # x = tf.layers.dense(x, units=self.z_dim * 8 * 8, name='dec-fc-1')
             x = tf.reshape(x, [-1, 8, 8, self.z_dim])
 
             for i in range(1, repeat + 1):

diff --git a/BEGAN/began_train.py b/BEGAN/began_train.py
@@ -44,8 +44,8 @@ def main():
         s.run(tf.global_variables_initializer())
 
         # Celeb-A DataSet images
-        ds = DataSet(input_height=32,
-                     input_width=32,
+        ds = DataSet(input_height=64,
+                     input_width=64,
                      input_channel=3,
                      mode='r').images
         dataset_iter = DataIterator(ds, None, train_step['batch_size'],