From 8ab2d26efed8c39a2df43b259e61fca0fc00bea6 Mon Sep 17 00:00:00 2001
From: Xinwei He <1154646635@qq.com>
Date: Sat, 1 Apr 2017 09:52:08 +0800
Subject: [PATCH] de-normalize images for displaying

sorry, the previous pull request  inserting real_cpu in the wrong place
---
 main.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/main.py b/main.py
index f097b01..7c3e638 100644
--- a/main.py
+++ b/main.py
@@ -218,8 +218,10 @@ for epoch in range(opt.niter):
             % (epoch, opt.niter, i, len(dataloader), gen_iterations,
             errD.data[0], errG.data[0], errD_real.data[0], errD_fake.data[0]))
         if gen_iterations % 500 == 0:
+            real_cpu = real_cpu.mul(0.5).add(0.5)
             vutils.save_image(real_cpu, '{0}/real_samples.png'.format(opt.experiment))
             fake = netG(Variable(fixed_noise, volatile=True))
+            fake.data = fake.data.mul(0.5).add(0.5)
             vutils.save_image(fake.data, '{0}/fake_samples_{1}.png'.format(opt.experiment, gen_iterations))
 
     # do checkpointing