Felix Marty commited on
Commit
36e2d86
1 Parent(s): 508744a
Files changed (1) hide show
  1. 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: {args.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
 
 
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