fixed typo inside the convert_onxx file, changed from "interference" to "inference"
All checks were successful
RIA Hub Workflow Demo / ria-demo (push) Successful in 40s
All checks were successful
RIA Hub Workflow Demo / ria-demo (push) Successful in 40s
This commit is contained in:
parent
34d2431bf0
commit
0e8b897912
|
@ -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 = "interference_recognition_model.ckpt"
|
model_checkpoint = "inference_recognition_model.ckpt"
|
||||||
|
|
||||||
print("Converting to ONNX...")
|
print("Converting to ONNX...")
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue
Block a user