lvwerra HF staff commited on
Commit
53e52ca
1 Parent(s): a082eb0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -51,12 +51,13 @@ for kn in kns:
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], marker="x", c="red", label="You are here!")
55
- plt.scatter([1.0], [0.0], marker="x", c="blue", label="Chinchilla optimal")
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")
 
60
  ax.yaxis.set_minor_locator(MultipleLocator(10))
61
 
62
  return fig
 
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=20, marker="x", c="red", label="You are here!")
55
+ plt.scatter([1.0], [0.0], marker="x", s=20, c="blue", label="Chinchilla optimal")
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")
60
+ plt.grid(True, which="minor", alpha=0.5)
61
  ax.yaxis.set_minor_locator(MultipleLocator(10))
62
 
63
  return fig