From 0a787e0a830e1b39cbde2b224ef09755cfa79d91 Mon Sep 17 00:00:00 2001 From: Marat Dukhan Date: Wed, 14 Dec 2016 14:38:12 -0500 Subject: [PATCH] Fix time per dataset in benchmark --- test/preprocess-bench.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/test/preprocess-bench.py b/test/preprocess-bench.py index 5728c0d2627..85599362a73 100644 --- a/test/preprocess-bench.py +++ b/test/preprocess-bench.py @@ -44,7 +44,7 @@ batch = next(train_iter) end_time = timer() print("Performance: {dataset:.0f} minutes/dataset, {batch:.2f} secs/batch, {image:.2f} ms/image".format( - dataset=(end_time - start_time) * len(train_loader) / (batch_count * args.batchSize) / 60.0, + dataset=(end_time - start_time) * (float(len(train_loader)) / batch_count / 60.0), batch=(end_time - start_time) / float(batch_count), image=(end_time - start_time) / (batch_count * args.batchSize) * 1.0e+3))