diff --git a/nbs/04_learner.ipynb b/nbs/04_learner.ipynb index 4fdc5a8..b8de429 100755 --- 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",