Skip to content

Instantly share code, notes, and snippets.

@skeeet

skeeet/mlSpeedTests.swift

Forked from akirasosa/mlSpeedTests.swift
Created May 19, 2020 09:40
Show Gist options

  • Select an option

    Learn more about clone URLs
  • Save skeeet/191944c46695aada2f9c69e8131e40d4 to your computer and use it in GitHub Desktop.

Select an option

Learn more about clone URLs
Save skeeet/191944c46695aada2f9c69e8131e40d4 to your computer and use it in GitHub Desktop.
Download ZIP

Revisions

  1. akirasosa created this gist Nov 8, 2017.
    32 changes: 32 additions & 0 deletions mlSpeedTests.swift
    View file
    Original file line number Diff line number Diff line change
    @@ -0,0 +1,32 @@
    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)
    }
    }
    }