fiddle with fig options
Browse files
app.py
CHANGED
@@ -145,7 +145,7 @@ def get_plot(model_name, generate_type):
|
|
145 |
df["framework"] = ["PyTorch", "TF (Eager Execition)", "TF (XLA)"]
|
146 |
df = pd.melt(df, id_vars=["framework"], value_vars=["T4", "3090", "A100"])
|
147 |
|
148 |
-
fig = plt.figure(figsize=(100, 6), dpi=200)
|
149 |
g = sns.catplot(
|
150 |
data=df, kind="bar",
|
151 |
x="variable", y="value", hue="framework",
|
@@ -153,7 +153,6 @@ def get_plot(model_name, generate_type):
|
|
153 |
)
|
154 |
g.despine(left=True)
|
155 |
g.set_axis_labels("GPU", "Generation time (ms)")
|
156 |
-
g.set(title=f"{generate_type} generation time for {model_name} (no bar means OOM)")
|
157 |
g.legend.set_title("Framework")
|
158 |
return plt.gcf()
|
159 |
|
|
|
145 |
df["framework"] = ["PyTorch", "TF (Eager Execition)", "TF (XLA)"]
|
146 |
df = pd.melt(df, id_vars=["framework"], value_vars=["T4", "3090", "A100"])
|
147 |
|
148 |
+
# fig = plt.figure(figsize=(100, 6), dpi=200)
|
149 |
g = sns.catplot(
|
150 |
data=df, kind="bar",
|
151 |
x="variable", y="value", hue="framework",
|
|
|
153 |
)
|
154 |
g.despine(left=True)
|
155 |
g.set_axis_labels("GPU", "Generation time (ms)")
|
|
|
156 |
g.legend.set_title("Framework")
|
157 |
return plt.gcf()
|
158 |
|