Skip to content

Commit f8a0365

Browse files
committed
better logs
1 parent f86a93d commit f8a0365

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

optimum/exporters/onnx/convert.py

+4-2
Original file line numberDiff line numberDiff line change
@@ -258,7 +258,7 @@ def _run_validation(
258258

259259
model_kwargs = model_kwargs if model_kwargs is not None else {}
260260

261-
logger.info(f"Validating ONNX model {onnx_model.as_posix()}...")
261+
logger.info(f"\nValidating ONNX model {onnx_model.as_posix()}...")
262262

263263
if atol is None:
264264
atol = config.ATOL_FOR_VALIDATION
@@ -764,7 +764,9 @@ def export_models(
764764
output_path = output_dir / output_name
765765
output_path.parent.mkdir(parents=True, exist_ok=True)
766766

767-
logger.info(f"Exporting submodel {i}/{len(models_and_onnx_configs)}: {submodel.__class__.__name__}")
767+
logger.info(
768+
f"\n***** Exporting submodel {i}/{len(models_and_onnx_configs)}: {submodel.__class__.__name__} *****"
769+
)
768770
outputs.append(
769771
export(
770772
model=submodel,

0 commit comments

Comments
 (0)