apple / apple/coremltools

Slow model loading

Open
#1,814 9 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

Hey,
I have been using coremltools for some time and model loading has been very fast, usually within seconds. I am using an ~150Mbs size mlpackage model with coremltool version 5.2.0. Recently, I upgraded to 6.2 and after this, the coreml model loading increased to 150secs.
Could anyone who faced a similar issue at any time during development help me in this regard.
my code is:

```
import coremltools as ct

model = ct.models.MLModel(model_path)

```

Thanks in advance!

system is mac m1 air, 8Gb mem

Contributor guide

Open the contributing guide

Research direction

Start by reproducing the reported 150-second load with ct.models.MLModel(model_path) using the ~150 MB mlpackage on the stated Mac M1 environment, comparing coremltools 5.2.0 and 6.2. Inspect the model-loading path for the version-related regression. Done means the slowdown is reproducible, its cause is identified, and the affected behavior has a regression test or documented resolution.

Written by the indexing model from the issue text.

Assessment

Tech stack
python
Domain
machine-learning, performance
Issue type
Bug
Difficulty
4/5
Estimated time
3-5 days
Activity status
Stale
Clarity
Needs clarification
Newbie friendliness
35/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.