apple / apple/coremltools

Missing Pytorch operators nn.MaxPool2d and nn.MaxUnpool2d

Open
#1,507 1 comment 1 reaction 0 assignees View on GitHub
missing layer type PyTorch (traced) triaged
Dominant language
Python
Stars
5.4k
Forks
850
Avg merge
4d 5h
Merged PRs (30d)
10

Description

_- Name of layer type:_ MaxPool2d, MaxUnpool2d
_- Is this a PyTorch or a TensorFlow layer type:_ Pytorch
_- Your version of coremltools:_ 5.2.0
_- Your version of PyTorch/TensorFlow:_ 1.11
_- Impact of supporting this layer type. Why is adding support for this layer type important? Is it necessary to support a popular model or use case?_
Nowadays, a lot of models use these layers, especially encoder/decoder ones. Recent state-of-the-art models cannot be converted to CoreML due to the lack of support for these layers

Contributor guide

Open the contributing guide

Research direction

The issue names the PyTorch MaxPool2d and MaxUnpool2d layer types but mentions no files, tests, or entry points. Start by locating the PyTorch operator-conversion support and existing pooling tests; done means models using both layers convert successfully to Core ML.

Written by the indexing model from the issue text.

Assessment

Tech stack
python, pytorch
Domain
machine-learning, tooling
Issue type
Feature
Difficulty
4/5
Estimated time
3-5 days
Activity status
Stale
Clarity
Needs clarification
Newbie friendliness
25/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.