Program crashes while trying to load onnx model

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.4k stars 534 forks source link

Program crashes while trying to load onnx model #90

Open MADHAVAN001 opened 4 years ago

MADHAVAN001 commented 4 years ago

This behavior is noted in convert_to_caffe2_models.py while trying to load a converted onnx model. The readme documentation was followed up until this point.

model = onnx.load(model_path) init_net, predict_net = c2.onnx_graph_to_caffe2_net(model)

mylyu commented 4 years ago

to be specific, it actually crashes on init_net, predict_net = c2.onnx_graph_to_caffe2_net(model)