apple / apple/coremltools

NeuralNetwork and MLprogram lead to very different results

Open
#2,301 12 comments 4 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

Hello! I'm using coremltools to convert a custom architecture from pytorch to coreML. I've noticed that I only get correct results if I use the neuralnetwork backend and not the mlprogram backend. I'm wondering if anyone has any high(or low) level insights as to why this might be the case? And where I might be able to start tracing the differences in representations to find the supposed bug? Assuming this is a bug

Contributor guide

Open the contributing guide

Research direction

The issue names no files, tests, or reproducible model. Start by reproducing the custom PyTorch conversion with both the neuralnetwork and mlprogram backends, then compare their generated representations. Done means isolating a confirmed conversion discrepancy and documenting a minimal reproduction.

Written by the indexing model from the issue text.

Assessment

Tech stack
python, pytorch
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.