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

reverted .onnx to .cptx

Browse Source
This commit is contained in:
Liyu Xiao 2025-05-22 15:04:19 -04:00
parent 81154ec85e
commit c6ad74b5df
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__": if __name__ == "__main__":
from checkpoint_files import CHECKPOINTS_DIR from checkpoint_files import CHECKPOINTS_DIR
model_checkpoint = "inference_recognition_model.onnx" model_checkpoint = "inference_recognition_model.ckpt"
print("Converting to ONNX...") print("Converting to ONNX...")