ERROR WHILE RUNNING train.py

[Karansheth]

Traceback (most recent call last): File "/content/drive/My Drive/Colab Notebooks/train.py", line 167, in enc_out = encoder(batch) File "/usr/local/lib/python3.6/dist-packages/torch/nn/modules/module.py", line 722, in _call_impl result = self.forward(*input, **kwargs) File "/content/drive/My Drive/Colab Notebooks/encoders/lf.py", line 92, in forward hist_embed = self.hist_rnn(hist_embed, batch['hist_len']) File "/usr/local/lib/python3.6/dist-packages/torch/nn/modules/module.py", line 722, in _call_impl result = self.forward(*input, **kwargs) File "/content/drive/My Drive/Colab Notebooks/utils/dynamic_rnn.py", line 34, in forward sorted_seq_input, lengths=sorted_len, batch_first=True) File "/usr/local/lib/python3.6/dist-packages/torch/nn/utils/rnn.py", line 234, in pack_padded_sequence lengths = torch.as_tensor(lengths, dtype=torch.int64) RuntimeError: CUDA error: device-side assert triggered ERROR WHILE RUNNING train.py