Example #1
0
 def test_make_z_normal(self):
     """Tests the function that makes the latent variable tensors."""
     if tf.executing_eagerly():
         return
     zs = generator.make_z_normal(2, 16, 128)
     sess = tf.train.MonitoredTrainingSession()
     z_batch = sess.run(zs)
     self.assertEqual((2, 16, 128), z_batch.shape)
Example #2
0
 def _make_noise(_):
     noise = gen_module.make_z_normal(num_towers, bs, params['z_dim'])
     return noise[0]  # one tower