set noisev to Volatile to freeze netG when updating netD.

2017-03-11 23:04:05 +08:00 · 2017-03-11 23:04:05 +08:00 · 59a48456ef
parent 2cf43baf21
commit 59a48456ef
1 changed files with 2 additions and 3 deletions
--- 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