cnzzx / GeMap

[ECCV'24] Online Vectorized HD Map Construction using Geometry
https://invictus717.github.io/GeMap/
Apache License 2.0
196 stars 16 forks source link

export onnx error ( We don't have an op for aten::mul but it isn't a special case ) #8

Open reskipper opened 5 months ago

reskipper commented 5 months ago

when i export GeMap to onnx ,report error as below : RuntimeError: 0INTERNAL ASSERT FAILED at "../torch/csrc/jit/ir/alias_analysis.cpp":532, please report a bug to PyTorch. We don't have an op for aten::mul but it isn't a special case. Argument types: Tensor, bool,

I also tried PyTorch 1.10.0, and encountered the same error. Any ideas on how to tackle this? Thanks a bunch!

cnzzx commented 5 months ago

Thanks for your interest. We didn't try to export GeMap to onnx, and not sure which operations are not supported. I guess it will be a tough job, but maybe this repo is valuable?

By the way, this repo is GeMap, not TopoNet :)

reskipper commented 5 months ago

Thank you for your response!

Sorry, I actually encountered the same error message when exporting gemap and toponet onnx models, which led to a typo. Apologies again.