Skip to content

Commit

Permalink
Fix latent
Browse files Browse the repository at this point in the history
  • Loading branch information
preritj authored Jan 26, 2018
1 parent 0acbc58 commit 04351d6
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions model.py
Original file line number Diff line number Diff line change
Expand Up @@ -177,7 +177,7 @@ def train(self):
# batch_gen = image_loader.batch_generator()

for i in range(self.cfg.n_iters):
batch_z = np.random.normal(-1, 1, size=(batch_size, z_dim))
batch_z = np.random.normal(0, 1, size=(batch_size, z_dim))
feed_dict = {self.tf_placeholders['z']: batch_z,
self.tf_placeholders['learning_rate']: learning_rate,
self.tf_placeholders['alpha']: alpha}
Expand Down Expand Up @@ -227,7 +227,7 @@ def generate_images(self, model, batch_z=None, alpha=0.):
batch_size = 64 # self.cfg.batch_size
z_dim = self.cfg.z_dim
if batch_z is None:
batch_z = np.random.normal(-1, 1, size=(batch_size, z_dim))
batch_z = np.random.normal(0, 1, size=(batch_size, z_dim))
# saver = tf.train.Saver(self.ema_vars)
saver = tf.train.Saver()
feed_dict = {self.tf_placeholders['z']: batch_z,
Expand Down

0 comments on commit 04351d6

Please sign in to comment.