跑模型时出现问题

[nishi123]

您好，我在跑您的网络时出现了这种错误（我的环境是python3.5，pytorch 0.4.1）
File "main.py", line 20, in
trainer.train()
File "/opt/share0/luchen/MGN/MGN-pytorch-master/trainer.py", line 49, in train
loss = self.loss(outputs, labels)
File "/usr/local/lib/python3.5/dist-packages/torch/nn/modules/module.py", line 491, in call
result = self.forward(*input, **kwargs)
File "/opt/share0/luchen/MGN/MGN-pytorch-master/loss/init.py", line 70, in forward
self.log[-1, i] += effective_loss.item()
RuntimeError: cuda runtime error (59) : device-side assert triggered at /pytorch/aten/src/THC/generic/THCStorage.c:36

辛苦您帮忙解答一下，谢谢！