The shape of the mask [32, 6375] at index 0 does not match the shape of the indexed tensor [204000, 1] at index 0

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

The shape of the mask [32, 6375] at index 0 does not match the shape of the indexed tensor [204000, 1] at index 0 #61

Closed xxlei closed 6 years ago

xxlei commented 6 years ago

File "/home/lei/PytorchSSD/layers/modules/refine_multibox_loss.py", line 112, in forward loss_c[pos] = 0 # filter out pos boxes for now RuntimeError: The shape of the mask [32, 6375] at index 0 does not match the shape of the indexed tensor [204000, 1] at index 0

WangTianYuan commented 5 years ago

@xxlei 你好，我遇到了你一样的问题，请问怎么解决的？

csm-kr commented 5 years ago

me too!