Commit 59d82d13 authored by Yuxin Wu's avatar Yuxin Wu

correct tensor name in WGAN example

parent b6a4f429
......@@ -34,7 +34,7 @@ class Model(DCGAN.Model):
def build_losses(self, vecpos, vecneg):
# the Wasserstein-GAN losses
self.d_loss = tf.reduce_mean(vecneg - vecpos, name='d_loss')
self.g_loss = -tf.reduce_mean(vecneg, name='g_loss')
self.g_loss = tf.negative(tf.reduce_mean(vecneg), name='g_loss')
add_moving_summary(self.d_loss, self.g_loss)
def _get_optimizer(self):
......
......@@ -114,7 +114,7 @@ def get_tqdm_kwargs(**kwargs):
if f.isatty():
default['mininterval'] = 0.5
else:
default['mininterval'] = 60
default['mininterval'] = 300
default.update(kwargs)
return default
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment