lvwerra HF staff commited on
Commit
13461c8
1 Parent(s): f05765d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
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=40, marker="x", c="red", label="You are here!")
55
- plt.scatter([1.0], [0.0], marker="x", s=40, c="blue", label="Chinchilla optimal")
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")