C:/w/b/windows/pytorch/aten/src/THC/THCTensorIndex.cu:272: block: [178,0,0], thread: [31,0,0] Assertion srcIndex < srcSelectDimSize failed.
Traceback (most recent call last):
File “C:/Users/lhuang93/PycharmProjects/pythonProject/train.py”, line 190, in 
F = train(config, ‘ddi_e-5.log’)
File “C:/Users/lhuang93/PycharmProjects/pythonProject/train.py”, line 130, in train
train_loss, train_pred = run_iter(batch=train_batch, is_training=True)
File “C:/Users/lhuang93/PycharmProjects/pythonProject/train.py”, line 102, in run_iter
logits = model(input_ids, attention_mask, token_type_ids, label, e1_mask, e2_mask)
File “D:\Anaconda\Anaconda\lib\site-packages\torch\nn\modules\module.py”, line 722, in _call_impl
result = self.forward(*input, **kwargs)
File “C:\Users\lhuang93\PycharmProjects\pythonProject\model.py”, line 60, in forward
outputs = self.bert(input_ids, attention_mask=attention_mask,
File “D:\Anaconda\Anaconda\lib\site-packages\torch\nn\modules\module.py”, line 722, in _call_impl
result = self.forward(*input, **kwargs)
File “D:\Anaconda\Anaconda\lib\site-packages\transformers\modeling_bert.py”, line 829, in forward
embedding_output = self.embeddings(
File “D:\Anaconda\Anaconda\lib\site-packages\torch\nn\modules\module.py”, line 722, in _call_impl
result = self.forward(*input, **kwargs)
File “D:\Anaconda\Anaconda\lib\site-packages\transformers\modeling_bert.py”, line 211, in forward
token_type_embeddings = self.token_type_embeddings(token_type_ids)
File “D:\Anaconda\Anaconda\lib\site-packages\torch\nn\modules\module.py”, line 722, in _call_impl
result = self.forward(*input, **kwargs)
File “D:\Anaconda\Anaconda\lib\site-packages\torch\nn\modules\sparse.py”, line 124, in forward
return F.embedding(
File “D:\Anaconda\Anaconda\lib\site-packages\torch\nn\functional.py”, line 1814, in embedding
return torch.embedding(weight, input, padding_idx, scale_grad_by_freq, sparse)
RuntimeError: CUDA error: device-side assert triggered