fix: 13_convolutions.ipynb - we should be normalizing activations (#366)

2021-01-17 21:11:30 +03:00
parent 829b68ca05
commit b4eac5e921
1 changed files with 1 additions and 1 deletions
--- a/13_convolutions.ipynb
+++ b/13_convolutions.ipynb
@@ -3415,8 +3415,8 @@
   "source": [
    "def conv(ni, nf, ks=3, act=True):\n",
    "    layers = [nn.Conv2d(ni, nf, stride=2, kernel_size=ks, padding=ks//2)]\n",
-    "    layers.append(nn.BatchNorm2d(nf))\n",
    "    if act: layers.append(nn.ReLU())\n",
+    "    layers.append(nn.BatchNorm2d(nf))\n",
    "    return nn.Sequential(*layers)"
   ]
  },