hassonofer commited on
Commit
6636eb2
·
1 Parent(s): b315517

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -2
app.py CHANGED
@@ -79,7 +79,7 @@ def plot_acc_memory(memory_compare_results_df: pl.DataFrame, width: int = 900, h
79
 
80
  chart = base + text + frontier
81
  return chart.properties(
82
- title=f"Accuray vs GPU Memory (batch size={batch_size}, amp={amp}", width=width, height=height
83
  ).configure_scale(zero=False)
84
 
85
 
@@ -290,7 +290,10 @@ def app() -> None:
290
 
291
  # Launch the app
292
  if __name__ == "__main__":
 
293
  for p in Path.glob(Path("."), "results_*.csv"):
294
- DATASETS.append(p.stem.removeprefix("results_"))
 
 
295
 
296
  app()
 
79
 
80
  chart = base + text + frontier
81
  return chart.properties(
82
+ title=f"Accuray vs GPU Memory (batch size={batch_size}, amp={amp})", width=width, height=height
83
  ).configure_scale(zero=False)
84
 
85
 
 
290
 
291
  # Launch the app
292
  if __name__ == "__main__":
293
+ file_info = []
294
  for p in Path.glob(Path("."), "results_*.csv"):
295
+ file_info.append((p.stat().st_size, p.stem.removeprefix("results_")))
296
+
297
+ DATASETS = [dataset_name for _, dataset_name in sorted(file_info, key=lambda x: x[1], reverse=True)]
298
 
299
  app()