apple / apple/coremltools

torch.nn.ConvTranspose1d Conversion failed

Open
#1,946 6 comments 0 reactions 1 assignee Claimed by @TobyRoseman View on GitHub
bug PyTorch (traced) triaged
Dominant language
Python
Stars
5.4k
Forks
850
Avg merge
4d 5h
Merged PRs (30d)
10

Description

## 🐞Describing the bug
Conversion for torch.nn.ConvTranspose1d failed

## Stack Trace
```
Traceback (most recent call last):
File "/Users/cvv/Downloads/Pycharm_project/StyleTTS/edge/coreml_test.py", line 9, in
mlprogram = ct.convert(
File "/Users/cvv/opt/anaconda3/envs/diffusion/lib/python3.8/site-packages/coremltools/converters/_converters_entry.py", line 530, in convert
mlmodel = mil_convert(
File "/Users/cvv/opt/anaconda3/envs/diffusion/lib/python3.8/site-packages/coremltools/converters/mil/converter.py", line 188, in mil_convert
return _mil_convert(model, convert_from, convert_to, ConverterRegistry, MLModel, compute_units, **kwargs)
File "/Users/cvv/opt/anaconda3/envs/diffusion/lib/python3.8/site-packages/coremltools/converters/mil/converter.py", line 212, in _mil_convert
proto, mil_program = mil_convert_to_proto(
File "/Users/cvv/opt/anaconda3/envs/diffusion/lib/python3.8/site-packages/coremltools/converters/mil/converter.py", line 286, in mil_convert_to_proto
prog = frontend_converter(model, **kwargs)
File "/Users/cvv/opt/anaconda3/envs/diffusion/lib/python3.8/site-packages/coremltools/converters/mil/converter.py", line 108, in __call__
return load(*args, **kwargs)
File "/Users/cvv/opt/anaconda3/envs/diffusion/lib/python3.8/site-packages/coremltools/converters/mil/frontend/torch/load.py", line 63, in load
return _perform_torch_convert(converter, debug)
File "/Users/cvv/opt/anaconda3/envs/diffusion/lib/python3.8/site-packages/coremltools/converters/mil/frontend/torch/load.py", line 102, in _perform_torch_convert
prog = converter.convert()
File "/Users/cvv/opt/anaconda3/envs/diffusion/lib/python3.8/site-packages/coremltools/converters/mil/frontend/torch/converter.py", line 439, in convert
convert_nodes(self.context, self.graph)
File "/Users/cvv/opt/anaconda3/envs/diffusion/lib/python3.8/site-packages/coremltools/converters/mil/frontend/torch/ops.py", line 92, in convert_nodes
add_op(context, node)
File "/Users/cvv/opt/anaconda3/envs/diffusion/lib/python3.8/site-packages/coremltools/converters/mil/frontend/torch/ops.py", line 943, in _convolution
conv = mb.slice_by_index(
File "/Users/cvv/opt/anaconda3/envs/diffusion/lib/python3.8/site-packages/coremltools/converters/mil/mil/ops/registry.py", line 183, in add_op
return cls._add_op(op_cls_to_add, **kwargs)
File "/Users/cvv/opt/anaconda3/envs/diffusion/lib/python3.8/site-packages/coremltools/converters/mil/mil/builder.py", line 182, in _add_op
new_op.type_value_inference()
File "/Users/cvv/opt/anaconda3/envs/diffusion/lib/python3.8/site-packages/coremltools/converters/mil/mil/operation.py", line 253, in type_value_inference
output_types = self.type_inference()
File "/Users/cvv/opt/anaconda3/envs/diffusion/lib/python3.8/site-packages/coremltools/converters/mil/mil/ops/defs/iOS15/tensor_transformation.py", line 531, in type_inference
ret_shape = solve_slice_by_index_shape(x_shape, begin, end, stride, begin_mask, end_mask, squeeze_mask)
File "/Users/cvv/opt/anaconda3/envs/diffusion/lib/python3.8/site-packages/coremltools/converters/mil/mil/ops/defs/_utils.py", line 473, in solve_slice_by_index_shape
raise ValueError(
ValueError: slice_by_index op: unsupported values in for dimension 2, (begin, end, stride) : (1, 0, 1)
```
## To Reproduce
```python
import torch
import torch.nn as nn
import coremltools as ct

pool = nn.ConvTranspose1d(512, 512, kernel_size=3, stride=2, groups=512, padding=1, output_padding=1).eval()
input = torch.rand((1, 512, 528))
jit_model = torch.jit.trace(pool, (input))

mlprogram = ct.convert(
jit_model,
inputs=[ct.TensorType(name='input', shape=input.shape)],
outputs=[ct.TensorType(name="output")],
source="pytorch",
convert_to="mlprogram",
compute_precision=ct.precision.FLOAT16,
)
```
model conversion failed.
## System environment (please complete the following information):
coremltools version: 7.0b1
OS: MacOS Ventura 13.1 (22C65)
torch: 1.12.1

Contributor guide

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.