qoherent/modrec-workflow
14
0
Fork 1
Code Issues 1 Pull Requests Workflows Packages Projects Releases 1 Wiki Activity

reverted .onnx to .cptx
All checks were successful
RIA Hub Workflow Demo / ria-demo (push) Successful in 40s
Details

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__":
from checkpoint_files import CHECKPOINTS_DIR
model_checkpoint = "inference_recognition_model.onnx"
model_checkpoint = "inference_recognition_model.ckpt"
print("Converting to ONNX...")