Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -56,7 +56,7 @@ def plot_acc_param(param_compare_results_df: pl.DataFrame, width: int = 1000, he
|
|
56 |
)
|
57 |
|
58 |
chart = base + text + frontier
|
59 |
-
return chart.properties(title="
|
60 |
|
61 |
|
62 |
def plot_acc_memory(memory_compare_results_df: pl.DataFrame, width: int = 900, height: int = 640) -> alt.LayerChart:
|
@@ -105,7 +105,7 @@ def plot_acc_memory(memory_compare_results_df: pl.DataFrame, width: int = 900, h
|
|
105 |
|
106 |
chart = base + text + frontier
|
107 |
return chart.properties(
|
108 |
-
title=f"
|
109 |
).configure_scale(zero=False)
|
110 |
|
111 |
|
|
|
56 |
)
|
57 |
|
58 |
chart = base + text + frontier
|
59 |
+
return chart.properties(title="Accuracy vs Parameter Count", width=width, height=height).configure_scale(zero=False)
|
60 |
|
61 |
|
62 |
def plot_acc_memory(memory_compare_results_df: pl.DataFrame, width: int = 900, height: int = 640) -> alt.LayerChart:
|
|
|
105 |
|
106 |
chart = base + text + frontier
|
107 |
return chart.properties(
|
108 |
+
title=f"Accuracy vs GPU Memory (batch size={batch_size}, amp={amp})", width=width, height=height
|
109 |
).configure_scale(zero=False)
|
110 |
|
111 |
|