From 4445eaa470ff321413b114c258de1be64d7156a6 Mon Sep 17 00:00:00 2001 From: Ross Wightman Date: Fri, 5 Mar 2021 16:48:31 -0800 Subject: [PATCH] Add img_size to benchmark output --- benchmark.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/benchmark.py b/benchmark.py index faac1fc5..e692eacc 100755 --- a/benchmark.py +++ b/benchmark.py @@ -208,6 +208,7 @@ class InferenceBenchmarkRunner(BenchmarkRunner): samples_per_sec=round(num_samples / t_run_elapsed, 2), step_time=round(1000 * total_step / num_samples, 3), batch_size=self.batch_size, + img_size=self.input_size[-1], param_count=round(self.param_count / 1e6, 2), ) @@ -310,6 +311,7 @@ class TrainBenchmarkRunner(BenchmarkRunner): bwd_time=round(1000 * total_bwd / num_samples, 3), opt_time=round(1000 * total_opt / num_samples, 3), batch_size=self.batch_size, + img_size=self.input_size[-1], param_count=round(self.param_count / 1e6, 2), ) else: