Compatibility with Pytorch 0.4

[saba155]

I am trying to run code using python 3.5 and I have pytorch 0.4.1 installed. I always end up getting error below: Traceback (most recent call last): File "train.py", line 205, in hidden_list = repackage_hidden(hidden_list) File "train.py", line 94, in repackage_hidden return tuple(repackage_hidden(v) for v in h) File "train.py", line 94, in return tuple(repackage_hidden(v) for v in h) File "train.py", line 94, in repackage_hidden return tuple(repackage_hidden(v) for v in h) File "train.py", line 94, in return tuple(repackage_hidden(v) for v in h) File "train.py", line 94, in repackage_hidden return tuple(repackage_hidden(v) for v in h) File "train.py", line 94, in return tuple(repackage_hidden(v) for v in h) File "train.py", line 94, in repackage_hidden return tuple(repackage_hidden(v) for v in h) File "train.py", line 94, in return tuple(repackage_hidden(v) for v in h) File "train.py", line 94, in repackage_hidden return tuple(repackage_hidden(v) for v in h) File "/home/usr/.local/lib/python3.5/site-packages/torch/tensor.py", line 381, in iter raise TypeError('iteration over a 0-d tensor') TypeError: iteration over a 0-d tensor

[BenjiDayan]

@SabaMehmood15 , I initially had the same error (on python 2.7, pytorch 0.4.1), the readme does say they tested on pytorch 0.3.0 and python 2.7. After downgrading pytorch it worked fine for me.

[SmallHedgehog]

Modify 'type(h) == Variable' to 'type(h) == torch.Tensor'

#-- helper func for rnn units def repackage_hidden(h): """Wraps hidden states in new Variables, to detach them from their history.""" # if type(h) == Variable: # return Variable(h.data) if type(h) == torch.Tensor: return Variable(h.data) else: return tuple(repackage_hidden(v) for v in h)