Skip to content

Commit

Permalink
separate_outputs and export_hw_optimized added to printouts benchmarks
Browse files Browse the repository at this point in the history
  • Loading branch information
Mehrdad committed Jan 24, 2024
1 parent 7724a53 commit 502c8cc
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion ultralytics/utils/benchmarks.py
Original file line number Diff line number Diff line change
Expand Up @@ -142,7 +142,7 @@ def benchmark(
df = pd.DataFrame(y, columns=["Format", "Status❔", "Size (MB)", key, "Inference time (ms/im)"])

name = Path(model.ckpt_path).name
s = f"\nBenchmarks complete for {name} on {data} at imgsz={imgsz} ({time.time() - t0:.2f}s)\n{df}\n"
s = f"\nBenchmarks complete for {name} on {data} at imgsz={imgsz}, separate_outputs={separate_outputs}, export_hw_optimized={export_hw_optimized} ({time.time() - t0:.2f}s)\n{df}\n"
LOGGER.info(s)
with open("benchmarks.log", "a", errors="ignore", encoding="utf-8") as f:
f.write(s)
Expand Down

0 comments on commit 502c8cc

Please sign in to comment.