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

export coreML not working #253

Open
DanielZanchi opened this issue Oct 17, 2023 · 0 comments
Open

export coreML not working #253

DanielZanchi opened this issue Oct 17, 2023 · 0 comments

Comments

@DanielZanchi
Copy link

I am trying to obtain a coreML model with an aspect ration of 1:1.
so I am using this command:
python3 export_coreml.py \ --model-variant mobilenetv3 \ --checkpoint rvm_mobilenetv3.pth \ --resolution 1080 1080 \ --downsample-ratio 0.25 \ --quantize-nbits 16 \ --output video_matting.mlmodel

But i get this error:
Traceback (most recent call last): File "/Users/danielzanchi/Downloads/RobustVideoMatting-coreml/export_coreml.py", line 192, in <module> Exporter() File "/Users/danielzanchi/Downloads/RobustVideoMatting-coreml/export_coreml.py", line 28, in __init__ self.register_custom_ops() File "/Users/danielzanchi/Downloads/RobustVideoMatting-coreml/export_coreml.py", line 57, in register_custom_ops def hardswish_(context, node): File "/usr/local/lib/python3.9/site-packages/coremltools/converters/mil/frontend/torch/torch_op_registry.py", line 36, in func_wrapper raise Exception( Exception: Attempting to register "hardswish_" op. Do not register inplace ops. (inplace torch ops end in a "_"). Instead register the normal op version: "hardswish". The inplace version will be supported automatically.

suggestions?

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