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

Convert models to ONNX #30

Open charmowen opened 5 years ago

charmowen commented 5 years ago

RuntimeError: ONNX export failed: Couldn't export operator aten::max_pool2d_with_indices?

qfgaohao commented 5 years ago

https://github.com/qfgaohao/pytorch-ssd/issues/33#issuecomment-467533485