search

lzx1413 / PytorchSSD

pytorch version of SSD and it's enhanced methods such as RFBSSD,FSSD and RefineDet
MIT License
709 stars 237 forks source link

RuntimeError: copy_if failed to synchronize: device-side assert triggered #97

Open Crystal950504 opened 4 years ago

Crystal950504 commented 4 years ago

File "train_refinedet.py", line 289, in train() File "train_refinedet.py", line 203, in train odm_loss_l, odm_loss_c = odm_criterion(out, targets) File "/home/shaoyidu/Program/anaconda3/envs/pytorch/lib/python3.6/site-packages/torch/nn/modules/module.py", line 491, in call result = self.forward(*input, **kwargs) File "/home/shaoyidu/Program/LH/RefineDet.PyTorch/layers/modules/refinedet_multibox_loss.py", line 127, in forward loss_c[pos.view(-1,1)] = 0 # filter out pos boxes for now RuntimeError: copy_if failed to synchronize: device-side assert triggered