error

[zhangyiling0920]

Traceback (most recent call last):
File "/Users/zyl/Downloads/BBBRNN-master/main.py", line 249, in
train_loss, train_nll, train_kl = train()
File "/Users/zyl/Downloads/BBBRNN-master/main.py", line 187, in train
hidden = repackage_hidden(hidden)
File "/Users/zyl/Downloads/BBBRNN-master/main.py", line 131, in repackage_hidden
return tuple(repackage_hidden(v) for v in h)
File "/Users/zyl/Downloads/BBBRNN-master/main.py", line 131, in
return tuple(repackage_hidden(v) for v in h)
File "/Users/zyl/Downloads/BBBRNN-master/main.py", line 131, in repackage_hidden
return tuple(repackage_hidden(v) for v in h)
File "/Users/zyl/Downloads/BBBRNN-master/main.py", line 131, in
return tuple(repackage_hidden(v) for v in h)
File "/Users/zyl/Downloads/BBBRNN-master/main.py", line 131, in repackage_hidden
return tuple(repackage_hidden(v) for v in h)
File "/Users/zyl/Downloads/BBBRNN-master/main.py", line 131, in
return tuple(repackage_hidden(v) for v in h)
File "/Users/zyl/Downloads/BBBRNN-master/main.py", line 131, in repackage_hidden
return tuple(repackage_hidden(v) for v in h)
File "/Users/zyl/Downloads/BBBRNN-master/main.py", line 131, in
return tuple(repackage_hidden(v) for v in h)
File "/Users/zyl/Downloads/BBBRNN-master/main.py", line 131, in repackage_hidden
return tuple(repackage_hidden(v) for v in h)
File "/Users/zyl/opt/anaconda3/envs/pytorch/lib/python3.8/site-packages/torch/_tensor.py", line 915, in iter
raise TypeError("iteration over a 0-d tensor")
TypeError: iteration over a 0-d tensor