Bug fix for wandb callback

m-gopichand · Aug 23, 2024 · c40cb8d · c40cb8d
1 parent d53ed05
commit c40cb8d
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 2 deletions.
diff --git a/nbs/04_learner.ipynb b/nbs/04_learner.ipynb
@@ -674,6 +674,7 @@
     "    def __init__(self, model, dls=(0,), loss_func=F.mse_loss, lr=0.1, cbs=None, opt_func=optim.Adam):\n",
     "        cbs = fc.L(cbs)\n",
     "        fc.store_attr()\n",
+    "        self.iter_total = 0\n",
     "\n",
     "    @contextmanager\n",
     "    def cb_ctx(self, nm):\n",
@@ -696,6 +697,7 @@
     "                        self.backward()\n",
     "                        self.step()\n",
     "                        self.zero_grad()\n",
+    "                    self.iter_total += 1\n",
     "    \n",
     "    def fit(self, n_epochs=1, train=True, valid=True, cbs=None, lr=None):\n",
     "        cbs = fc.L(cbs)\n",

diff --git a/rapidai/__init__.py b/rapidai/__init__.py
@@ -1 +1 @@
-__version__ = "0.0.6"
+__version__ = "0.0.7"
diff --git a/settings.ini b/settings.ini
@@ -1,7 +1,7 @@
 [DEFAULT]
 repo = rapidai
 lib_name = rapidai
-version = 0.0.6
+version = 0.0.7
 min_python = 3.9
 license = apache2
 black_formatting = False