seathiefwang / MGN-pytorch

Reproduction of paper: Learning Discriminative Features with Multiple Granularities for Person Re-Identification
285 stars 108 forks source link

跑模型时出现问题 #9

Open nishi123 opened 6 years ago

nishi123 commented 6 years ago

您好,我在跑您的网络时出现了这种错误(我的环境是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

辛苦您帮忙解答一下,谢谢!