You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
I discovered that the expected number of arguments to cumsum op in coremltools might be wrong while exporting PyTorch module to *.pte file (executorch). See
Number of arguments is set to 3 while only two are used. Indeed, in my export, I got an error that cumsum op expects 3 arguments while only 2 were provided. Changing this value to 2 fixed suppressed the error.
To Reproduce
I'm not able to reproduce the error with small code sample.
System environment (please complete the following information):
coremltools version: 8.0b1
OS (e.g. MacOS version or Linux type): MacOS Version 14.5
Any other relevant version information (e.g. PyTorch or TensorFlow version): PyTorch 2.5.0.dev20240716
Additional context
Discovered this while exporting Pytorch model to executorch with CoreML backend.
The text was updated successfully, but these errors were encountered:
I discovered that the expected number of arguments to
cumsum
op in coremltools might be wrong while exporting PyTorch module to *.pte file (executorch). Seecoremltools/coremltools/converters/mil/frontend/torch/ops.py
Line 931 in 8b7048e
cumsum
op expects 3 arguments while only 2 were provided. Changing this value to 2 fixed suppressed the error.To Reproduce
I'm not able to reproduce the error with small code sample.
System environment (please complete the following information):
Additional context
Discovered this while exporting Pytorch model to executorch with CoreML backend.
The text was updated successfully, but these errors were encountered: