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')