Skip to content

GitLab

Commit e4c811c9 authored by Soumith Chintala's avatar Soumith Chintala Committed by GitHub
Browse files

Merge pull request #14 from Maratyszcza/master 

Fix time per dataset in benchmark
parents 038a01fe 0a787e0a
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
test/preprocess-bench.py
View file @ e4c811c9
...@@ -44,7 +44,7 @@ if __name__ == "__main__": ...@@ -44,7 +44,7 @@ if __name__ == "__main__":
batch = next(train_iter) batch = next(train_iter)
end_time = timer() end_time = timer()
print("Performance: {dataset:.0f} minutes/dataset, {batch:.2f} secs/batch, {image:.2f} ms/image".format( 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), batch=(end_time - start_time) / float(batch_count),
image=(end_time - start_time) / (batch_count * args.batchSize) * 1.0e+3)) image=(end_time - start_time) / (batch_count * args.batchSize) * 1.0e+3))
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment