fiddle with plots
Browse files
app.py
CHANGED
@@ -147,6 +147,7 @@ def get_plot(model_name, plot_eager, generate_type):
|
|
147 |
if plot_eager == "No":
|
148 |
df = df[df["framework"] != "TF (Eager Execition)"]
|
149 |
|
|
|
150 |
g = sns.catplot(
|
151 |
data=df,
|
152 |
kind="bar",
|
@@ -154,7 +155,7 @@ def get_plot(model_name, plot_eager, generate_type):
|
|
154 |
y="value",
|
155 |
hue="framework",
|
156 |
palette={"PyTorch": "blue", "TF (Eager Execition)": "orange", "TF (XLA)": "red"},
|
157 |
-
|
158 |
)
|
159 |
g.despine(left=True)
|
160 |
g.set_axis_labels("GPU", "Generation time (ms)")
|
|
|
147 |
if plot_eager == "No":
|
148 |
df = df[df["framework"] != "TF (Eager Execition)"]
|
149 |
|
150 |
+
plt.figure(dpi=1200)
|
151 |
g = sns.catplot(
|
152 |
data=df,
|
153 |
kind="bar",
|
|
|
155 |
y="value",
|
156 |
hue="framework",
|
157 |
palette={"PyTorch": "blue", "TF (Eager Execition)": "orange", "TF (XLA)": "red"},
|
158 |
+
alpha=.9,
|
159 |
)
|
160 |
g.despine(left=True)
|
161 |
g.set_axis_labels("GPU", "Generation time (ms)")
|