expected type torch.FloatTensor but got torch.cuda.FloatTensor
weixin_43981448:
Traceback (most recent call last):
File "run_bert_ner.py", line 5, in
start()
File "/root/bert_ner/main.py", line 31, in start
verbose=1)
File "/root/bert_ner/train/train.py", line 108, in fit
train_loss.backward()
File "/root/anaconda3/envs/python367/lib/python3.6/site-packages/torch/tensor.py", line 118, in backward
torch.autograd.backward(self, gradient, retain_graph, create_graph)
File "/root/anaconda3/envs/python367/lib/python3.6/site-packages/torch/autograd/__init__.py", line 93, in backward
allow_unreachable=True) # allow_unreachable flag
RuntimeError: Function AddBackward0 returned an invalid gradient at index 1 - expected type torch.cuda.FloatTensor but got torch.FloatTensor
请问这是什么原因?应该怎么改?
|