Pre-heat CoreML MLModel

user_4156957 0
Viewed 239

I'm seeing up to 3x-4x slower results on my first prediction on my MLModel. Seems like once you run the first prediction it's heating the model. Couldn't find any official information about it.

What is the right way to pre-heat my model? Call it on an async thread on app launch with dummy data?

mlmodel coreml swift ios
1 Answers

As @Alladinian mentioned, starting from iOS 14 you can use the MLModel.load(contentsOf:...) function.
Here is a usage example for pre-loading a local .mlmodel file:

if let url = Bundle.main.url(forResource: "myModel", withExtension: "mlmodelc") {
    let config = MLModelConfiguration()
    MLModel.load(contentsOf: url, configuration: config) { [weak self] result in
        switch result {
            case .success(let model):
                print("Model loaded and ready.")
                let modelWrapper = MyModelWrapper(model: model)
            case .failure(let error):
                print("Error loading model: \(error)")
        }
    }
}

If you want to pre-load an externally fetched model, make sure it's compiled using MLModel.compileModel(at:).

user_41272580
Related