paul-corbalan
/
wasserstein-gan
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix mroe bugs

This commit is contained in:
Soumith Chintala 2017-02-28 08:30:25 -05:00
parent d9ad2bb847
commit 7eee65dc9c
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
main.py
View File

@ -178,7 +178,8 @@ for epoch in range(opt.niter):
real_cpu, _ = data real_cpu, _ = data
netD.zero_grad() netD.zero_grad()
batch_size = real_cpu.size(0) batch_size = real_cpu.size(0)
input.resize_as_(real_cpu).copy_(real_cpu)
inputv = Variable(input) inputv = Variable(input)
errD_real = netD(inputv) errD_real = netD(inputv)