Skip to content

Commit 0a787e0

Browse files
committed
Fix time per dataset in benchmark
1 parent 038a01f commit 0a787e0

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

test/preprocess-bench.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -44,7 +44,7 @@
4444
batch = next(train_iter)
4545
end_time = timer()
4646
print("Performance: {dataset:.0f} minutes/dataset, {batch:.2f} secs/batch, {image:.2f} ms/image".format(
47-
dataset=(end_time - start_time) * len(train_loader) / (batch_count * args.batchSize) / 60.0,
47+
dataset=(end_time - start_time) * (float(len(train_loader)) / batch_count / 60.0),
4848
batch=(end_time - start_time) / float(batch_count),
4949
image=(end_time - start_time) / (batch_count * args.batchSize) * 1.0e+3))
5050

0 commit comments

Comments
 (0)