caogang / wgan-gp

A pytorch implementation of Paper "Improved Training of Wasserstein GANs"
MIT License
1.51k stars 345 forks source link

an error occurred #23

Open lzqcode opened 7 years ago

lzqcode commented 7 years ago

Hello,when I run gan_cifar10.py, an error occurred: Traceback (most recent call last): File "gan_cifar10.py", line 187, in _data = gen.next() File "gan_cifar10.py", line 170, in inf_train_gen for images, target in train_gen(): ValueError: too many values to unpack, I don't know how this mistake happened

js-fan commented 6 years ago

Just change this line (170) into: for images in train_gen():

Also, line 258, for images, _ in dev_gen(): should be changed into: for images in dev_gen():