We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
2 parents 038a01f + 0a787e0 commit e4c811cCopy full SHA for e4c811c
test/preprocess-bench.py
@@ -44,7 +44,7 @@
44
batch = next(train_iter)
45
end_time = timer()
46
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,
+ dataset=(end_time - start_time) * (float(len(train_loader)) / batch_count / 60.0),
48
batch=(end_time - start_time) / float(batch_count),
49
image=(end_time - start_time) / (batch_count * args.batchSize) * 1.0e+3))
50
0 commit comments