fill in the other model sizes

dagelf · May 26, 2024 · ee6fa5e · ee6fa5e
1 parent 5ec4b77
commit ee6fa5e
Showing 1 changed file with 7 additions and 7 deletions.
diff --git 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",