Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -50,9 +50,9 @@ for kn in kns:
|
|
50 |
|
51 |
def plot_curve(kn, kd):
|
52 |
fig, ax = plt.subplots()
|
53 |
-
plt.plot(kns, overheads, color="black")
|
54 |
-
plt.scatter([kn], [compute_overhead(kn, kd)*100], s=
|
55 |
-
plt.scatter([1.0], [0.0], marker="x", s=
|
56 |
plt.xlabel("Fraction of compute optimal model size")
|
57 |
plt.ylabel("Compute overhead (%)")
|
58 |
plt.legend(loc="best")
|
|
|
50 |
|
51 |
def plot_curve(kn, kd):
|
52 |
fig, ax = plt.subplots()
|
53 |
+
plt.plot(kns, overheads, color="black", zorder=1)
|
54 |
+
plt.scatter([kn], [compute_overhead(kn, kd)*100], s=64, marker="x", c="red", label="You are here!", zorder=2)
|
55 |
+
plt.scatter([1.0], [0.0], marker="x", s=64, c="blue", label="Chinchilla optimal", zorder=2)
|
56 |
plt.xlabel("Fraction of compute optimal model size")
|
57 |
plt.ylabel("Compute overhead (%)")
|
58 |
plt.legend(loc="best")
|