Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Any documentation on how to add more layer types? #1990

Open
joshterrill opened this issue Jun 13, 2024 · 1 comment
Open

Any documentation on how to add more layer types? #1990

joshterrill opened this issue Jun 13, 2024 · 1 comment

Comments

@joshterrill
Copy link

1. 使用环境(environment)

  • OS: Mac
  • OS Version: Sonoma 14.2.1

2. Github版本

  • branch:master

3. 详细描述bug 情况 (Describe the bug)
A clear and concise description of what the bug is.

  • issue type: convert failed
  • original model: CoreML (espresso file)

4. 日志(Log)

expand_dims
Error: unsupported layer type: expand_dims

Is there any sort of documentation around that I can reference if I want to add additional supported layer types to tools/onnx2tnn/onnx-coreml/coreml2onnx.py?

@joshterrill
Copy link
Author

joshterrill commented Jun 13, 2024

I figured out how to modify the existing coreml2onnx script to support blob_weights_f16 and softmax. There were also a few bugs that I fixed where the shapes of some of the conv layers that were being generated were incorrect. Would you take a PR to look at it?

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant