ch/Caps-Train
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix continuation error

Browse Source
This commit is contained in:
Christoph Hagen 2023-10-24 10:38:13 +02:00
parent 874688b43f
commit 2913878b3e
2 changed files with 11 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
Sources/ClassifierCreator.swift
View File

@ -293,7 +293,7 @@ final class ClassifierCreator {
parameters: params,
sessionParameters: sessionParameters)
} catch {
throw TrainingError.failedToCreateClassifier("\(error)")
throw TrainingError.failedToCreateClassifier(error)
}
job.progress
@ -301,7 +301,6 @@ final class ClassifierCreator {
.sink { completed in
Swift.print(String(format: " %.1f %% completed", completed * 100), terminator: "\r")
fflush(stdout)
//guard let progress = MLProgress(progress: job.progress) else {
// return
//}
@ -314,7 +313,12 @@ final class ClassifierCreator {
return try await withCheckedThrowingContinuation { continuation in
// Register a sink to receive the resulting model.
job.result.sink { result in
continuation.resume(throwing: TrainingError.failedToCreateClassifier("\(result)"))
switch result {
case .finished:
break // Continuation already called with model
case .failure(let error):
continuation.resume(throwing: TrainingError.failedToCreateClassifier(error))
}
} receiveValue: { [weak self] model in
// Use model
self?.print(info: "Created model")
@ -334,12 +338,11 @@ final class ClassifierCreator {
trainingData: .labeledDirectories(at: imageDirectory),
parameters: params)
} catch {
throw TrainingError.failedToCreateClassifier("\(error)")
throw TrainingError.failedToCreateClassifier(error)
}
}
private func save(model: MLImageClassifier) throws {
print(info: "Saving classifier...")
do {
try model.write(to: classifierUrl)

6
Sources/TrainingError.swift
View File

@ -31,7 +31,7 @@ enum TrainingError: Error {
case failedToGetClassifierVersion(Error)
case invalidClassifierVersion(String)
case failedToCreateClassifier(String)
case failedToCreateClassifier(Error)
case failedToWriteClassifier(Error)
case failedToReadClassifierData(Error)
@ -95,8 +95,8 @@ extension TrainingError: CustomStringConvertible {
return "Failed to get classifier version: \(error)"
case .invalidClassifierVersion(let string):
return "Invalid classifier version \(string)"
case .failedToCreateClassifier(let result):
return "Failed to create classifier: \(result)"
case .failedToCreateClassifier(let error):
return "Failed to create classifier: \(error)"
case .failedToWriteClassifier(let error):
return "Failed to save model to file: \(error)"
case .failedToReadClassifierData(let error):