model_hash_name + base_name

Signed-off-by: jcwchen <[email protected]>
onnx · Jun 21, 2023 · 902d751 · 902d751
1 parent 8b0fb00
commit 902d751
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/workflow_scripts/run_mlagility.py b/workflow_scripts/run_mlagility.py
@@ -60,7 +60,7 @@ def main():
             mlagility_created_onnx = osp.join(cache_converted_dir, model_hash_name, "onnx", model_hash_name + base_name)
             if args.create:
                 ort_test_dir_utils.create_test_dir(mlagility_created_onnx, "./", final_model_dir)
-                rename(osp.join(final_model_dir, model_hash_name), final_model_path)
+                rename(osp.join(final_model_dir, model_hash_name + base_name), final_model_path)
                 print(f"Successfully created {model_zoo_dir} by mlagility and ORT.")
             else:
                 shutil.copy(mlagility_created_onnx, final_model_path)