RuntimeError: set_storage_offset is not allowed on Tensor created from .data or .detach()
Opened this issue · 1 comments
jrothschild33 commented
- 问题1:
Traceback (most recent call last):
File "/cnn-text-classification-pytorch/main.py", line 112, in <module>
train.train(train_iter, dev_iter, cnn, args)
File "/cnn-text-classification-pytorch/train.py", line 25, in train
feature.data.t_(), target.data.sub_(1) # batch first, index align
RuntimeError: set_storage_offset is not allowed on Tensor created from .data or .detach()
Process finished with exit code 1
- 问题1解决:将【2处】
feature.data.t_(), target.data.sub_(1)
替换为:
feature = feature.data.t()
target = target.data.sub(1)
- 问题2:
Traceback (most recent call last):
File "/cnn-text-classification-pytorch/main.py", line 112, in <module>
train.train(train_iter, dev_iter, cnn, args)
File "/cnn-text-classification-pytorch/train.py", line 43, in train
loss.data[0],
IndexError: invalid index of a 0-dim tensor. Use tensor.item() to convert a 0-dim tensor to a Python number
Process finished with exit code 1
- 问题2解决:将【2处】
loss.data[0]
替换为:loss.item()
inkyusa commented
Hope this helps!
inkyusa@6dfefe7