From 7eee65dc9cd08414dfcd5718b5cbe7b4c0360495 Mon Sep 17 00:00:00 2001 From: Soumith Chintala Date: Tue, 28 Feb 2017 08:30:25 -0500 Subject: [PATCH] fix mroe bugs --- main.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/main.py b/main.py index 30f3b47..b943b62 100644 --- 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)