From 59a48456ef03d35df2ff53f3fc51f63a5cb6fdd5 Mon Sep 17 00:00:00 2001 From: chenyuntc Date: Sat, 11 Mar 2017 23:04:05 +0800 Subject: [PATCH] set noisev to Volatile to freeze netG when updating netD. --- main.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/main.py b/main.py index 5db5ab0..f097b01 100644 --- a/main.py +++ b/main.py @@ -190,10 +190,9 @@ for epoch in range(opt.niter): # train with fake noise.resize_(opt.batchSize, nz, 1, 1).normal_(0, 1) - noisev = Variable(noise) - fake = netG(noisev) + noisev = Variable(noise, volatile = True) # totally freeze netG + fake = Variable(netG(noisev).data) inputv = fake - inputv.detach() errD_fake = netD(inputv) errD_fake.backward(mone) errD = errD_real - errD_fake