examples: improve pytorch microbenchmark

eric-haibin-lin · Jul 23, 2019 · 59dbbd6 · 59dbbd6
1 parent 00ab3bf
commit 59dbbd6
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/example/pytorch/microbenchmark-byteps.py b/example/pytorch/microbenchmark-byteps.py
@@ -61,8 +61,8 @@ def benchmark(tensor, average, name):
 # Benchmark
 log('Running benchmark...')
 
-log('size (Byte)    avg. time (ms)    std.dev (ms)')
-for i in range(10):
+log('size (Byte) \t avg. time (ms) \t std.dev (ms)')
+for i in range(8):
     size = 10**i
     data = torch.rand(size, dtype=torch.float32)
     if args.cuda:
@@ -78,6 +78,6 @@ def benchmark(tensor, average, name):
     avg = np.mean(durations)
     std = np.std(durations)
 
-    log('%d    %s    %s' % (4*size, '%.3f'%avg, '%.3f'%std))
+    log('%d \t %s \t %s' % (4*size, '%.3f'%avg, '%.3f'%std))
 
 log('End benchmark.')