From 81154ec85e2ddd4cb6268f10121a6ab4f7abc127 Mon Sep 17 00:00:00 2001 From: Liyu Xiao Date: Thu, 22 May 2025 15:03:04 -0400 Subject: [PATCH] tweaked typo inside convert_to_onnx --- convert_to_onnx.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/convert_to_onnx.py b/convert_to_onnx.py index 4616290..a6b6ca7 100644 --- a/convert_to_onnx.py +++ b/convert_to_onnx.py @@ -72,7 +72,7 @@ def convert_to_onnx(ckpt_path, fp16=False): if __name__ == "__main__": from checkpoint_files import CHECKPOINTS_DIR - model_checkpoint = "inference_recognition_model.ckpt" + model_checkpoint = "inference_recognition_model.onnx" print("Converting to ONNX...")