From e58713d550efa0d2e8d4dc6b061723f9ae6cb7c4 Mon Sep 17 00:00:00 2001
From: Sergey Zagoruyko <zagoruyko2@gmail.com>
Date: Sat, 10 Dec 2016 23:41:13 +0100
Subject: [PATCH] Correct total batch num in dcgan

after https://github.com/pytorch/pytorch/pull/277
---
 dcgan/main.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/dcgan/main.py b/dcgan/main.py
index a183a2c0f2..fbac678052 100644
--- a/dcgan/main.py
+++ b/dcgan/main.py
@@ -227,7 +227,7 @@ def forward(self, input):
         optimizerG.step()
 
         print('[%d/%d][%d/%d] Loss_D: %f Loss_G: %f'
-              % (epoch, opt.niter, i, len(dataloader)/opt.batchSize,
+              % (epoch, opt.niter, i, len(dataloader),
                  errD.data[0], errG.data[0]))
         if i % 100 == 0:
             vutils.save_image(real_cpu, 'real_samples.png')