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

fix bugs

This commit is contained in:
Soumith Chintala 2017-02-27 14:41:08 -05:00
parent e553093d3b
commit 93f26f9e4c
1 changed files with 0 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
main.py
View File

@ -139,10 +139,6 @@ if opt.cuda:
one, mone = one.cuda(), mone.cuda() one, mone = one.cuda(), mone.cuda()
noise, fixed_noise = noise.cuda(), fixed_noise.cuda() noise, fixed_noise = noise.cuda(), fixed_noise.cuda()
input = Variable(input)
noise = Variable(noise)
fixed_noise = Variable(fixed_noise)
# setup optimizer # setup optimizer
if opt.adam: if opt.adam:
optimizerD = optim.Adam(netD.parameters(), lr=opt.lrD, betas=(opt.beta1, 0.999)) optimizerD = optim.Adam(netD.parameters(), lr=opt.lrD, betas=(opt.beta1, 0.999))