search

ShuangXieIrene / ssds.pytorch

Repository for Single Shot MultiBox Detector and its variants, implemented with pytorch, python3.
MIT License
571 stars 166 forks source link

When I train model, get this problem! #31

Open RobotIntelligence opened 5 years ago

RobotIntelligence commented 5 years ago

File "ssds.pytorch/lib/layers/modules/multibox_loss.py", line 91, in forward loss_c[pos] = 0 # filter out pos boxes for now RuntimeError: The shape of the mask [32, 11620] at index 0 does not match the shape of the indexed tensor [371840, 1] at index 0

jeffreycbw commented 5 years ago

Hi,

You can refer to the thread here, hope it helps.