fix bugs
This commit is contained in:
parent
e553093d3b
commit
93f26f9e4c
4
main.py
4
main.py
|
@ -139,10 +139,6 @@ if opt.cuda:
|
|||
one, mone = one.cuda(), mone.cuda()
|
||||
noise, fixed_noise = noise.cuda(), fixed_noise.cuda()
|
||||
|
||||
input = Variable(input)
|
||||
noise = Variable(noise)
|
||||
fixed_noise = Variable(fixed_noise)
|
||||
|
||||
# setup optimizer
|
||||
if opt.adam:
|
||||
optimizerD = optim.Adam(netD.parameters(), lr=opt.lrD, betas=(opt.beta1, 0.999))
|
||||
|
|
Loading…
Reference in New Issue