Skip to content

Commit 2186177

Browse files
committed
chore: minor fixes
Signed-off-by: Dheeraj Peri <[email protected]>
1 parent d5dbc4d commit 2186177

File tree

1 file changed

+3
-4
lines changed

1 file changed

+3
-4
lines changed

tools/perf/perf_run.py

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -89,8 +89,7 @@ def run_torch_tensorrt(model, input_tensors, params, precision, truncate_long_an
8989
if precision == "int8":
9090
compile_settings.update({"calib": params.get("calibration_cache")})
9191

92-
with torchtrt.logging.debug():
93-
model = torchtrt.compile(model, **compile_settings)
92+
model = torchtrt.compile(model, **compile_settings)
9493

9594
iters = params.get("iterations", 20)
9695
# Warm up
@@ -292,8 +291,8 @@ def recordStats(backend, timings, precision, batch_size=1):
292291
"Batch size": batch_size,
293292
"Median(FPS)": speed_med,
294293
"Mean(FPS)": speed_mean,
295-
"Median-Latency(ms)": time_med,
296-
"Mean-Latency(ms)": time_mean,
294+
"Median-Latency(ms)": time_med*1000,
295+
"Mean-Latency(ms)": time_mean*1000,
297296
}
298297
results.append(stats)
299298

0 commit comments

Comments
 (0)