From ee6fa5ec9fe385a399bcaf264f720dabbbf72373 Mon Sep 17 00:00:00 2001
From: Andrej Karpathy <andrej.karpathy@gmail.com>
Date: Sun, 26 May 2024 18:04:43 +0000
Subject: [PATCH] fill in the other model sizes

---
 dev/vislog.ipynb | 14 +++++++-------
 1 file changed, 7 insertions(+), 7 deletions(-)

diff --git a/dev/vislog.ipynb b/dev/vislog.ipynb
index c753831f5..983f8fde1 100644
--- a/dev/vislog.ipynb
+++ b/dev/vislog.ipynb
@@ -51,15 +51,15 @@
     "sz = \"124M\"\n",
     "loss_baseline = {\n",
     "    \"124M\": 3.424958,\n",
-    "    \"350M\": None,\n",
-    "    \"774M\": None,\n",
-    "    \"1558M\": None,\n",
+    "    \"350M\": 3.083089,\n",
+    "    \"774M\": 3.000580,\n",
+    "    \"1558M\": 2.831273,\n",
     "}[sz]\n",
     "hella_baseline = {\n",
-    "    \"124M\": 0.2955,\n",
-    "    \"350M\": None,\n",
-    "    \"774M\": None,\n",
-    "    \"1558M\": None,\n",
+    "    \"124M\": 0.294463,\n",
+    "    \"350M\": 0.375224,\n",
+    "    \"774M\": 0.431986,\n",
+    "    \"1558M\": 0.488946,\n",
     "}[sz]\n",
     "\n",
     "# assumes each model run is stored in this way\n",