shape not match

[seefun]

PyramidBox/layers/modules/multibox_loss.py", line 105, in forward
loss_c[pos] = 0 # filter out pos boxes for now
RuntimeError: The shape of the mask [4, 34125] at index 0 does not match the shape of the indexed tensor [136500, 1] at index 0

I set the batchsize to 4 and I use my own dataset. When i run train.py I occur this problem. why does it happen?

[xxlei]

hei, i meet the same problem, do you sloved it ?

[abcdvzz]

hei, i meet the same problem, do you sloved it ?

Did u solve the problem?

[abcdvzz]

PyramidBox/layers/modules/multibox_loss.py", line 105, in forward
loss_c[pos] = 0 # filter out pos boxes for now
RuntimeError: The shape of the mask [4, 34125] at index 0 does not match the shape of the indexed tensor [136500, 1] at index 0

I set the batchsize to 4 and I use my own dataset. When i run train.py I occur this problem. why does it happen?

Did u solve the problem?

[Tianlock]

@seefun hello, have you solved the problem? i met the same problem.

[Tianlock]

@xxlei Hi, have you solved it, i met the same problem, thanks

[seefun]

@abcdvzz @Tianlock I give up using this codes. And I try to use the official pyramidbox codes which you can find in PaddlePaddle/models

[NarcissusInMirror]

I think this repo is modified on the pytorch version of SSD, the original repo has the related issue. All you need is to change few lines of code in multibox_loss.py and train.py as described in
this and this issue. It works well for me.