search

ShuangXieIrene / ssds.pytorch

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

RuntimeErroe: The size if tensor a(4) must match the size of tensor b(2) at non-singleton dimension #59

Closed wilxy closed 5 years ago

wilxy commented 5 years ago

When I use my own dataset and weight to detect picture, this error occurs: RuntimeErroe: The size if tensor a(4) must match the size of tensor b(2) at non-singleton dimension

wilxy commented 5 years ago

My yml file is OK when trainning, but have the above error when detect

wilxy commented 5 years ago

I got it! If you use torch 0.4.0 to train and 0.3.1 to test, the error will occur. Just use 0.3.1 to train and test!