From c40cb8d6363856d823cfca97142282a32ae81264 Mon Sep 17 00:00:00 2001 From: Gopichand Madala Date: Fri, 23 Aug 2024 22:55:25 +0530 Subject: [PATCH] Bug fix for wandb callback --- nbs/04_learner.ipynb | 2 ++ rapidai/__init__.py | 2 +- settings.ini | 2 +- 3 files changed, 4 insertions(+), 2 deletions(-) diff --git a/nbs/04_learner.ipynb b/nbs/04_learner.ipynb index 31a51ca..4734044 100755 --- 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 index 034f46c..7d53ea3 100755 --- a/rapidai/__init__.py +++ b/rapidai/__init__.py @@ -1 +1 @@ -__version__ = "0.0.6" +__version__ = "0.0.7" \ No newline at end of file diff --git a/settings.ini b/settings.ini index 07054bf..19d37ca 100755 --- 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