Update app.py

app.py

@@ -51,8 +51,8 @@ for kn in kns:
 def plot_curve(kn, kd):
     fig, ax = plt.subplots()
     plt.plot(kns, overheads, color="black", zorder=1)
-    plt.scatter([kn], [compute_overhead(kn, kd)*100], s=100, marker="x", c="red", label="You are here!", zorder=2)
-    plt.scatter([1.0], [0.0], marker="x", s=100, c="blue", label="Chinchilla optimal", zorder=2)
+    plt.scatter([kn], [compute_overhead(kn, kd)*100], s=100, marker="o", c="red", label="You are here!", zorder=2)
+    plt.scatter([1.0], [0.0], marker="o", s=100, c="blue", label="Chinchilla optimal", zorder=2)
     plt.xlabel("Fraction of Chinchilla optimal model size")
     plt.ylabel("Compute overhead (%)")
     plt.legend(loc="best")
@@ -71,8 +71,6 @@ def compute(N, D):
 
     kn = Bn*N/N_opt
     kd = compute_kd(kn)
-
-    print(N, D, N_opt/Bn, D_opt/Bn, kn, kd)
     
     fig = plot_curve(kn, kd)