您好,我在跑您的网络时出现了这种错误(我的环境是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
您好,我在跑您的网络时出现了这种错误(我的环境是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
辛苦您帮忙解答一下,谢谢!