lvwerra HF staff commited on
Commit
4110cfc
1 Parent(s): c73cf94

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -51,6 +51,8 @@ def plot_curve(kn, kd):
51
  fig = plt.figure()
52
  plt.plot(kns, overheads, color="black")
53
  plt.scatter([kn], [compute_overhead(kn, kd)*100], marker="x", c="red", label="You are here!")
 
 
54
  plt.xlabel("Fraction of compute optimal model size")
55
  plt.ylabel("Compute overhead (%)")
56
  plt.legend(loc="best")
 
51
  fig = plt.figure()
52
  plt.plot(kns, overheads, color="black")
53
  plt.scatter([kn], [compute_overhead(kn, kd)*100], marker="x", c="red", label="You are here!")
54
+ plt.scatter([1.0], [100.0], marker="x", c="blue", label="Chinchilla optimal")
55
+ plt.grid(True)
56
  plt.xlabel("Fraction of compute optimal model size")
57
  plt.ylabel("Compute overhead (%)")
58
  plt.legend(loc="best")