IndexError: invalid index of a 0-dim tensor

[cuongcx]

Run command: python main.py onto -lstm_type single -goal onto -enhanced_mention

Error:

.../model_utils.py:167: UserWarning: Implicit dimension choice for softmax has been deprecated. Change the call to include dim=X as an argument.
weighted_keys = self.key_softmax(k).view(input_embed.size()[0], -1, 1)
Traceback (most recent call last):
File "main.py", line 276, in
_train(config)
File "main.py", line 133, in _train
total_loss += loss.data.cpu()[0]
IndexError: invalid index of a 0-dim tensor. Use tensor.item() to convert a 0-dim tensor to a Python number

[helloleejq]

Run command: python main.py onto -lstm_type single -goal onto -enhanced_mention

Error:

.../model_utils.py:167: UserWarning: Implicit dimension choice for softmax has been deprecated. Change the call to include dim=X as an argument.
weighted_keys = self.key_softmax(k).view(input_embed.size()[0], -1, 1)
Traceback (most recent call last):
File "main.py", line 276, in
_train(config)
File "main.py", line 133, in _train
total_loss += loss.data.cpu()[0]
IndexError: invalid index of a 0-dim tensor. Use tensor.item() to convert a 0-dim tensor to a Python number

total_loss += loss.data.cpu().item()