microsoft / onnxjs

ONNX.js: run ONNX models using JavaScript
Other
1.75k stars 130 forks source link

Error: invalid name for initializer: 1059 #250

Open FahriBilici opened 3 years ago

FahriBilici commented 3 years ago

Hello, I trained custom tiny yolov4 model and convert it into onnx model. I cloned the demo from GitHub and replace yolo model with my model. But when I run the application it gives me Error: invalid name for initializer: 1059. How can I fix that?

Sxela commented 3 years ago

I've tried exportig model with these keyword arguments:

keep_initializers_as_inputs=True,
                  export_params=True

and it helped