IlyasMoutawwakil HF staff commited on
Commit
6ea5c03
·
1 Parent(s): b71e276
Files changed (1) hide show
  1. run.py +3 -3
run.py CHANGED
@@ -45,7 +45,7 @@ def run_benchmark(kwargs):
45
  f"model={model}",
46
  f"device={device}",
47
  f"benchmark={benchmark}",
48
- f"experiment_name={experiment_name}_baseline",
49
  ]
50
  for component, value in kwargs.items():
51
  if f"{benchmark}." in component.label:
@@ -93,11 +93,11 @@ def run_benchmark(kwargs):
93
  return
94
 
95
  if baseline:
96
- baseline_table = pd.read_csv(f"runs/{experiment_name}_baseline/{benchmark}_results.csv", index_col=0)
97
  table = pd.read_csv(f"runs/{experiment_name}/{benchmark}_results.csv", index_col=0)
98
  # concat tables
99
  table = pd.concat([baseline_table, table], axis=0)
100
- table["experiment_name"] = [experiment_name + "_baseline", experiment_name]
101
  table = table.set_index("experiment_name")
102
  table.reset_index(inplace=True)
103
  # compute speedups
 
45
  f"model={model}",
46
  f"device={device}",
47
  f"benchmark={benchmark}",
48
+ f"experiment_name=baseline",
49
  ]
50
  for component, value in kwargs.items():
51
  if f"{benchmark}." in component.label:
 
93
  return
94
 
95
  if baseline:
96
+ baseline_table = pd.read_csv(f"runs/baseline/{benchmark}_results.csv", index_col=0)
97
  table = pd.read_csv(f"runs/{experiment_name}/{benchmark}_results.csv", index_col=0)
98
  # concat tables
99
  table = pd.concat([baseline_table, table], axis=0)
100
+ table["experiment_name"] = ["baseline", experiment_name]
101
  table = table.set_index("experiment_name")
102
  table.reset_index(inplace=True)
103
  # compute speedups