lvwerra HF staff commited on
Commit
3001b8c
1 Parent(s): 2cddfc6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -51,9 +51,9 @@ for kn in kns:
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")
59
  plt.grid(True, which="both")
 
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=100, marker="x", c="red", label="You are here!", zorder=2)
55
+ plt.scatter([1.0], [0.0], marker="x", s=100, c="blue", label="Chinchilla optimal", zorder=2)
56
+ plt.xlabel("Fraction of Chinchilla optimal model size")
57
  plt.ylabel("Compute overhead (%)")
58
  plt.legend(loc="best")
59
  plt.grid(True, which="both")