ZiJianZhao / SeqGAN-PyTorch

A implementation of SeqGAN in PyTorch, following the implementation in tensorflow.
261 stars 93 forks source link

Problem with running #8

Open dokato opened 5 years ago

dokato commented 5 years ago

Hi, thanks for the great code.

When I tried to run your vanilla example python main.py I got:

IndexError: invalid index of a 0-dim tensor. Use tensor.item() to convert a 0-dim tensor to a Python number

I'm running it under python 3.5.6 and pytorch 1.0.0

hija commented 5 years ago

Hey Dokato, you might want to have a look at my PullRequest which fixes this bug and some more :)