.

m-gopichand · Aug 24, 2024 · 19385bb · 19385bb
1 parent 3a9cb10
commit 19385bb
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/nbs/04_learner.ipynb b/nbs/04_learner.ipynb
@@ -1455,14 +1455,14 @@
     "        self.run.finish()\n",
     "\n",
     "\n",
-    "class EarlyStoppingCallback(Callback):\n",
+    "class EarlyStoppingCB(Callback):\n",
     "    order = MetricsCB.order+1\n",
     "    def __init__(self, patience=3, min_delta=0.001):\n",
     "        self.patience = patience\n",
     "        self.min_delta = min_delta\n",
     "        self.best_loss = float('inf')\n",
     "        self.wait = 0\n",
-    "        \n",
+    "\n",
     "    def after_epoch(self, learn):\n",
     "        if not learn.model.training:\n",
     "            current_loss = learn.metrics.all_metrics['loss'].compute()\n",