search

qfgaohao / pytorch-ssd

MobileNetV1, MobileNetV2, VGG based SSD/SSD-lite implementation in Pytorch 1.0 / Pytorch 0.4. Out-of-box support for retraining on Open Images dataset. ONNX and Caffe2 support. Experiment Ideas like CoordConv.
https://medium.com/@smallfishbigsea/understand-ssd-and-implement-your-own-caa3232cd6ad
MIT License
1.39k stars 529 forks source link

使用voc数据集改动模型结构报错 #97

Open amitabhama opened 4 years ago

amitabhama commented 4 years ago

IndexError: The shape of the mask [1, 3000] at index 1 does not match the shape of the indexed tensor [1, 3600] at index 1 改动了模型结构,但模型整体可以跑通,训练过程中报错, 请问下造成这个错误有可能是那些问题

andgitchang commented 4 years ago

Possibly you should also modify self.source_layer_indexes (GraphPath) after customizing the model.