diff --git a/miniai/init.py b/miniai/init.py index 7bbb7fe..e42d91c 100644 --- a/miniai/init.py +++ b/miniai/init.py @@ -106,7 +106,7 @@ def lsuv_init(model, m, m_in, xb): # %% ../nbs/11_initializing.ipynb 114 def conv(ni, nf, ks=3, stride=2, act=nn.ReLU, norm=None, bias=None): - if bias is None: bias = not isinstance(norm, (nn.BatchNorm1d,nn.BatchNorm2d,nn.BatchNorm3d)) + if bias is None: bias = not (norm in (nn.BatchNorm1d,nn.BatchNorm2d,nn.BatchNorm3d)) layers = [nn.Conv2d(ni, nf, stride=stride, kernel_size=ks, padding=ks//2, bias=bias)] if norm: layers.append(norm(nf)) if act: layers.append(act()) diff --git a/nbs/09_learner.ipynb b/nbs/09_learner.ipynb index db56dc0..d76c4c9 100644 --- a/nbs/09_learner.ipynb +++ b/nbs/09_learner.ipynb @@ -86,22 +86,7 @@ "execution_count": null, "id": "b22868a9", "metadata": {}, - "outputs": [ - { - "data": { - "application/vnd.jupyter.widget-view+json": { - "model_id": "25a1693df8844081b050b8bbdb5f00fa", - "version_major": 2, - "version_minor": 0 - }, - "text/plain": [ - " 0%| | 0/2 [00:00