Merge pull request #7 from kmader/patch-1

Fixing for progressbar 3.6.2
ikostrikov · Apr 11, 2016 · 44426b3 · 44426b3
2 parents 36b2730 + 1df221a
commit 44426b3
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/main-draw.py b/main-draw.py
@@ -315,7 +315,7 @@ def get_reconstruction_cost(output_tensor, target_tensor, epsilon=1e-8):
             training_loss = 0.0
 
             widgets = ["epoch #%d|" % epoch, Percentage(), Bar(), ETA()]
-            pbar = ProgressBar(FLAGS.updates_per_epoch, widgets=widgets)
+            pbar = ProgressBar(max_value = FLAGS.updates_per_epoch, widgets=widgets)
             pbar.start()
             for i in range(FLAGS.updates_per_epoch):
                 pbar.update(i)

diff --git a/main-gan.py b/main-gan.py
@@ -132,7 +132,7 @@ def get_generator_loss(D2):
             generator_loss = 0.0
 
             widgets = ["epoch #%d|" % epoch, Percentage(), Bar(), ETA()]
-            pbar = ProgressBar(FLAGS.updates_per_epoch, widgets=widgets)
+            pbar = ProgressBar(max_value = FLAGS.updates_per_epoch, widgets=widgets)
             pbar.start()
             for i in range(FLAGS.updates_per_epoch):
                 pbar.update(i)

diff --git a/main-vae.py b/main-vae.py
@@ -140,7 +140,7 @@ def get_reconstruction_cost(output_tensor, target_tensor, epsilon=1e-8):
             training_loss = 0.0
 
             widgets = ["epoch #%d|" % epoch, Percentage(), Bar(), ETA()]
-            pbar = ProgressBar(FLAGS.updates_per_epoch, widgets=widgets)
+            pbar = ProgressBar(max_value = FLAGS.updates_per_epoch, widgets=widgets)
             pbar.start()
             for i in range(FLAGS.updates_per_epoch):
                 pbar.update(i)