Hi, thanks for sharing the code.
Not sure what is the source of error, I am using Python 3.6 and PyTorch 0.4.
The line where the error occurred is _data = gen.next(), in the file gan_cifar10.py
# (1) Update D network
###########################
for p in netD.parameters(): # reset requires_grad
p.requires_grad = True # they are set to False below in netG update
for i in range(CRITIC_ITERS):
_data = gen.next()
netD.zero_grad()
Hi, thanks for sharing the code. Not sure what is the source of error, I am using Python 3.6 and PyTorch 0.4.
The line where the error occurred is
_data = gen.next()
, in the file gan_cifar10.py