Felix Marty
commited on
Commit
·
36e2d86
1
Parent(s):
508744a
typo
Browse files- onnx_export.py +1 -1
onnx_export.py
CHANGED
@@ -73,7 +73,7 @@ def convert_onnx(model_id: str, task: str, folder: str) -> List:
|
|
73 |
validate_model_outputs(onnx_config, model, output, onnx_outputs, atol)
|
74 |
print(f"All good, model saved at: {output}")
|
75 |
except ValueError:
|
76 |
-
print(f"An error occured, but the model was saved at: {
|
77 |
|
78 |
n_files = len([name for name in os.listdir(folder) if os.path.isfile(os.path.join(folder, name)) and not name.startswith(".")])
|
79 |
|
|
|
73 |
validate_model_outputs(onnx_config, model, output, onnx_outputs, atol)
|
74 |
print(f"All good, model saved at: {output}")
|
75 |
except ValueError:
|
76 |
+
print(f"An error occured, but the model was saved at: {output.as_posix()}")
|
77 |
|
78 |
n_files = len([name for name in os.listdir(folder) if os.path.isfile(os.path.join(folder, name)) and not name.startswith(".")])
|
79 |
|