fix mroe bugs

2017-02-28 08:30:25 -05:00 · 2017-02-28 08:30:25 -05:00 · 7eee65dc9c
commit 7eee65dc9c
parent d9ad2bb847
1 changed files with 2 additions and 1 deletions
--- a/main.py
+++ b/main.py
@ -179,6 +179,7 @@ for epoch in range(opt.niter):
            netD.zero_grad()
            batch_size = real_cpu.size(0)

+            input.resize_as_(real_cpu).copy_(real_cpu)
            inputv = Variable(input)

            errD_real = netD(inputv)