File tree Expand file tree Collapse file tree 2 files changed +4
-7
lines changed
examples/demo-apps/apple_ios/ExecuTorchDemo/ExecuTorchDemo/Sources/App Expand file tree Collapse file tree 2 files changed +4
-7
lines changed Original file line number Diff line number Diff line change @@ -11,21 +11,20 @@ import MobileNetClassifier
11
11
import SwiftUI
12
12
13
13
enum Mode : String , CaseIterable {
14
- case cpu = " CPU "
14
+ case xnnpack = " XNNPACK "
15
15
case coreML = " CoreML "
16
16
case mps = " MPS "
17
- case xnnpack = " XNNPACK "
18
17
}
19
18
20
19
class ClassificationController : ObservableObject {
21
- @AppStorage ( " mode " ) var mode : Mode = . cpu
20
+ @AppStorage ( " mode " ) var mode : Mode = . xnnpack
22
21
@Published var classifications : [ Classification ] = [ ]
23
22
@Published var elapsedTime : TimeInterval = 0.0
24
23
@Published var isRunning = false
25
24
26
25
private let queue = DispatchQueue ( label: " org.pytorch.executorch.demo " , qos: . userInitiated)
27
26
private var classifier : ImageClassification ?
28
- private var currentMode : Mode = . cpu
27
+ private var currentMode : Mode = . xnnpack
29
28
30
29
func classify( _ image: UIImage ) {
31
30
guard !isRunning else {
@@ -62,8 +61,6 @@ class ClassificationController: ObservableObject {
62
61
private func createClassifier( for mode: Mode ) throws -> ImageClassification ? {
63
62
let modelFileName : String
64
63
switch mode {
65
- case . cpu:
66
- modelFileName = " mv3 "
67
64
case . coreML:
68
65
modelFileName = " mv3_coreml "
69
66
case . mps:
Original file line number Diff line number Diff line change @@ -33,7 +33,7 @@ struct ContentView: View {
33
33
TopBar ( title: " ExecuTorch Demo " )
34
34
ClassificationLabelView ( controller: classificationController)
35
35
Spacer ( )
36
- ClassificationTimeView ( controller: classificationController)
36
+ ClassificationTimeView ( controller: classificationController) . hidden ( )
37
37
ModeSelector ( controller: classificationController)
38
38
}
39
39
}
You can’t perform that action at this time.
0 commit comments