Skip to content

Instantly share code, notes, and snippets.

Created November 8, 2017 11:37
  • Star 6 You must be signed in to star a gist
  • Fork 1 You must be signed in to fork a gist
Star You must be signed in to star a gist
Save akirasosa/0fe78b4c173aeb49d96076db20aa9dd2 to your computer and use it in GitHub Desktop.
Benchmark Core ML model in iOS.
import CoreML
import XCTest
@testable import mlsample
class mlsampleTests: XCTestCase {
override func setUp() {
override func 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)
} catch {
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