search

eriklindernoren / PyTorch-GAN

PyTorch implementations of Generative Adversarial Networks.
MIT License
16.41k stars 4.07k forks source link

wgan issuse #8

Closed Sophia-intel closed 6 years ago

Sophia-intel commented 6 years ago

when I run python3 wgan.py, appears print ("[Epoch %d/%d] [Batch %d/%d] [D loss: %f] [G loss: %f]" % (epoch, opt.n_epochs,batches_done % len(dataloader), len(dataloader),d_loss.item(), gen_validity.item())) ValueError: only one element tensors can be converted to Python scalars how can i fix it?

eriklindernoren commented 6 years ago

Hi. This is fixed in f37d0f2. Thanks!

Sophia-intel commented 6 years ago

@eriklindernoren hey, it really works.Thanks!