michael/modrec-workflow
1
0
Fork 0
forked from qoherent/modrec-workflow
Code Pull Requests Workflows Activity

tweaked typo inside convert_to_onnx

Browse Source
This commit is contained in:
Liyu Xiao 2025-05-22 15:03:04 -04:00
parent 0e8b897912
commit 81154ec85e
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
convert_to_onnx.py
View File

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