Open dokato opened 5 years ago
Hi, thanks for the great code.
When I tried to run your vanilla example python main.py I got:
python main.py
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
3.5.6
1.0.0
Hey Dokato, you might want to have a look at my PullRequest which fixes this bug and some more :)
Hi, thanks for the great code.
When I tried to run your vanilla example
python main.py
I got:I'm running it under python
3.5.6
and pytorch1.0.0