Merge pull request #68 from torchgan/ap/image-patch-1

Fix logging of incorrect image names
torchgan · Dec 15, 2018 · 6dd681e · 6dd681e
2 parents cb5b512 + 556ed27
commit 6dd681e
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/torchgan/logging/visualize.py b/torchgan/logging/visualize.py
@@ -212,4 +212,6 @@ def __call__(self, trainer, **kwargs):
                     image = generator(*self.test_noise[pos])
                     image = torchvision.utils.make_grid(image)
                     super(ImageVisualize, self).__call__(trainer, image, model, **kwargs)
+                self.step -= 1
                 pos = pos + 1
+        self.step += 1 if pos > 0 else 0