Pointing at bert-small

Finetune BERT.ipynb

@@ -463,8 +463,22 @@
     "[bert-base some hp tuning](https://huggingface.co/datasets/frugal-ai-challenge/public-leaderboard-text/blob/main/submissions/Nonnormalizable_20250120_231350.json):\\\n",
     "acc 0.707, energy 0.803 Wh\n",
     "\n",
-    "bert-tiny, Nvidia T4 small:\\\n",
-    "acc 0.618, energy 0.079 Wh"
+    "Added normal data loader, batch size 32. Moved to Nvidia T4 small.\n",
+    "\n",
+    "bert-tiny\\\n",
+    "acc 0.618, energy 0.079 Wh\n",
+    "\n",
+    "bert-mini\\\n",
+    "acc 0.650, energy 0.129 Wh\n",
+    "\n",
+    "bert-small\\\n",
+    "acc, energy\n",
+    "\n",
+    "bert-medium\\\n",
+    "acc, energy\n",
+    "\n",
+    "bert-base\\\n",
+    "acc, energy"
    ]
   },
   {

tasks/text.py

@@ -13,13 +13,13 @@ from .utils.emissions import tracker, clean_emissions_data, get_space_info
 
 router = APIRouter()
 
-MODEL_TYPE = "bert-mini"
+MODEL_TYPE = "bert-small"
 DESCRIPTIONS = {
     "baseline": "baseline most common class",
     "bert-base": "bert base fine tuned on just training data, Nvidia T4 small",
     "bert-medium": "bert medium fine tuned on just training data, Nvidia T4 small",
     "bert-small": "bert small fine tuned on just training data, Nvidia T4 small",
-    "bert-mini": "bert min fine tuned on just training data, Nvidia T4 small",
+    "bert-mini": "bert mini fine tuned on just training data, Nvidia T4 small",
     "bert-tiny": "bert tiny fine tuned on just training data, Nvidia T4 small",
 }
 ROUTE = "/text"