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...")