search

Unity-Technologies / barracuda-release

Other
564 stars 76 forks source link

Problem with importing model #336

Closed andreapulita closed 3 months ago

andreapulita commented 3 months ago

Hello, I created a model with PyTorch and converted it into an Onnx model, but when I try to import it in Unity I get this error: OnnxImportException: Unknown type torch_nn_modules_container_Sequential_linear_relu_stack_1 encountered while parsing layer linear_relu_stack_1. I can give the model if it is needed.

andreapulita commented 3 months ago

I solved it creating the model with pytorch onnx.export instead of the onnx.dynamo_export