Skip to content

Commit e4c811c

Browse files
authored
Merge pull request #14 from Maratyszcza/master
Fix time per dataset in benchmark
2 parents 038a01f + 0a787e0 commit e4c811c

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)