fix mroe bugs

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

            errD_real = netD(inputv)