apple / apple/coremltools

coremltools 6.2 has troubles to convert the layer of mvit_block_0_transformer_0_attention

Open
#1,796 5 comments 0 reactions 0 assignees View on GitHub
question
Dominant language
Python
Stars
5.4k
Forks
850
Avg merge
4d 5h
Merged PRs (30d)
10

Description

## ❓Question

Layer 55: mvit_block_0_transformer_0_LN1
Input shape: (None, 4, 1024, 144)
Output shape: (None, 4, 1024, 144)
Input spec: None
Layer 56: mvit_block_0_transformer_0_attention
Input shape: (None, 4, 1024, 144)
Output shape: (None, 4, 1024, 144)
Input spec: None

lib/python3.7/site-packages/coremltools/converters/mil/mil/operation.py", line 318, in type_value_inference
f'Core ML only supports tensors with rank <= 5. Layer "{self.name}", '
ValueError: Core ML only supports tensors with rank <= 5. Layer "reshape_16", with type "reshape", outputs a rank 6 tensor.

Any ideas?

Thanks a lot.

Contributor guide

Open the contributing guide

Research direction

Start with the traceback location in coremltools/converters/mil/mil/operation.py at line 318 and the reported reshape_16 operation. Reproduce the conversion using the supplied layer shapes, then identify the model and source framework involved. Done requires a minimal reproducible case and a confirmed resolution or documented limitation.

Written by the indexing model from the issue text.

Assessment

Tech stack
python
Domain
machine-learning
Issue type
Bug
Difficulty
5/5
Estimated time
Over a week
Activity status
Stale
Clarity
Needs clarification
Newbie friendliness
20/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.