Runtime Error when using F.normalize

THCudaCheck FAIL file=/opt/conda/conda-bld/pytorch_1532579245307/work/aten/src/THC/generated/…/generic/THCTensorMathReduce.cu line=219 error=59 : device-side assert triggered
Traceback (most recent call last):
File “main.py”, line 242, in
main(args)
File “main.py”, line 234, in main
train(args)
File “main.py”, line 117, in train
_, iter_loss = network.forward(labels, p1, p2)
File “/home/ubuntu/701-Project/code/models/model.py”, line 77, in forward
predicted = self.model(word_tensor1, word_tensor2, p1_orig_len, p2_orig_len)
File “/home/ubuntu/anaconda3/envs/pytorch_p36/lib/python3.6/site-packages/torch/nn/modules/module.py”, line 477, in call
result = self.forward(*input, **kwargs)
File “/home/ubuntu/701-Project/code/models/mixmatch.py”, line 135, in forward
p1_input = F.normalize(p1_input, 2, 2)
File “/home/ubuntu/anaconda3/envs/pytorch_p36/lib/python3.6/site-packages/torch/nn/functional.py”, line 2257, in normalize
return input / input.norm(p, dim, True).clamp(min=eps).expand_as(input)
RuntimeError: cuda runtime error (59) : device-side assert triggered at /opt/conda/conda-bld/pytorch_1532579245307/work/aten/src/THC/generated/…/generic/THCTensorMathReduce.cu:219