[DOCS] Improved plots in tutorials

2021-03-11 00:29:16 -05:00
parent eacbb73968
commit 50e58d73db
8 changed files with 122 additions and 82 deletions
--- a/python/bench/bench_cross_entropy.py
+++ b/python/bench/bench_cross_entropy.py
@@ -9,7 +9,6 @@ confs = [
              y_vals  = ['triton', 'torch'],
              y_lines = ['Triton', 'Torch'],
              ylabel  = 'GBPS',
-              loglog  = False,
              plot_name = f'{mode}-2048',
              args = {'M': 2048, 'dtype': torch.float16, 'mode': mode}
    )\