Skip to content

Instantly share code, notes, and snippets.

Embed
What would you like to do?
Benchmark Core ML model in iOS.
import CoreML
import XCTest
@testable import mlsample
class mlsampleTests: XCTestCase {
override func setUp() {
super.setUp()
}
override func tearDown() {
super.tearDown()
}
func testSpeed() {
guard let input_data = try? MLMultiArray(shape:[3, 160, 160], dataType:MLMultiArrayDataType.double) else {
fatalError("Unexpected runtime error. MLMultiArray")
}
let model = mobile_unet_160_100_100()
// It's for excluding 1st measurement.
do {
try model.prediction(data: input_data)
print("predicted------------")
} catch {
print("error------------")
}
print("--------started------------")
self.measure() {
try! model.prediction(data: input_data)
}
}
}
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment