From 80cafb831bf0f32d3f9b52c584f5cc6e122b936e Mon Sep 17 00:00:00 2001 From: cpprhtn Date: Mon, 12 Jun 2023 09:05:20 -0700 Subject: [PATCH] fix: typo error --- ViT4MNIST/ViT4MNIST/HandwrittenDigitRecogizer.swift | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ViT4MNIST/ViT4MNIST/HandwrittenDigitRecogizer.swift b/ViT4MNIST/ViT4MNIST/HandwrittenDigitRecogizer.swift index ad26a65..18cb217 100644 --- a/ViT4MNIST/ViT4MNIST/HandwrittenDigitRecogizer.swift +++ b/ViT4MNIST/ViT4MNIST/HandwrittenDigitRecogizer.swift @@ -17,7 +17,7 @@ class HandwrittenDigitRecognizer { let module = InferenceModule(fileAtPath: filePath) { return module } else { - fatalError("Failed to load model file vit4mnist.pth") + fatalError("Failed to load model file vit4mnist.ptl") } }()