Traceback (most recent call last):


 "/u/71/irwantc1/unix/.conda/envs/ANP_RNN_torch/lib/python3.8/site-packages/pytorch_lightning/trainer/logging.py", line 108, in process_output

if k not in ['progress_bar', 'log', 'hiddens']:
KeyError: 'loss'

Traceback (most recent call last):


 "/home/harman/harman_workspace/poincare-embeddings/train.py", line 19, in train_mp

train(model, data, optimizer, opt, log, rank, queue)


 "/home/harman/harman_workspace/poincare-embeddings/train.py", line 46, in train

for inputs, targets in loader:


 "/home/harman/anaconda3/envs/pytorch-venv/lib/python3.6/site-packages/torch/utils/data/dataloader.py", line 286, in __next__

return self._process_next_batch(batch)


 "/home/harman/anaconda3/envs/pytorch-venv/lib/python3.6/site-packages/torch/utils/data/dataloader.py", line 307, in _process_next_batch

raise batch.exc_type(batch.exc_msg)
KeyError: 'Traceback (most recent call last):\n

 "/home/harman/anaconda3/envs/pytorch-venv/lib/python3.6/site-packages/torch/utils/data/dataloader.py", line 57, in _worker_loop\n
samples = collate_fn([dataset[i] for i in batch_indices])\n

 "/home/harman/anaconda3/envs/pytorch-venv/lib/python3.6/site-packages/torch/utils/data/dataloader.py", line 57, in <listcomp>\n
samples = collate_fn([dataset[i] for i in batch_indices])\n

 "/home/harman/harman_workspace/poincare-embeddings/model.py", line 185, in __getitem__\n
if n not in self._weights[t]:\nKeyError: tensor(491)\n' 
