fix norm at stem

Author: ayasyrev

model_constructor/net.py

@@ -103,7 +103,7 @@ def body(self):
     def _make_stem(self):
         stem = [(f"conv_{i}", self.conv_layer(self.stem_sizes[i], self.stem_sizes[i+1],
                     stride=2 if i==0 else 1,
-                    norm=(not self.stem_bn_end) if i==(len(self.stem_sizes)-2) else True,
+                    bn_layer=(not self.stem_bn_end) if i==(len(self.stem_sizes)-2) else True,
                     act_fn=self.act_fn, bn_1st=self.bn_1st))
                 for i in range(len(self.stem_sizes)-1)]
         stem.append(('stem_pool', self.stem_pool))

nbs/04_Net.ipynb

@@ -392,7 +392,7 @@
     "    def _make_stem(self):\n",
     "        stem = [(f\"conv_{i}\", self.conv_layer(self.stem_sizes[i], self.stem_sizes[i+1], \n",
     "                    stride=2 if i==0 else 1, \n",
-    "                    norm=(not self.stem_bn_end) if i==(len(self.stem_sizes)-2) else True,\n",
+    "                    bn_layer=(not self.stem_bn_end) if i==(len(self.stem_sizes)-2) else True,\n",
     "                    act_fn=self.act_fn, bn_1st=self.bn_1st))\n",
     "                for i in range(len(self.stem_sizes)-1)]\n",
     "        stem.append(('stem_pool', self.stem_pool))\n",
@@ -1245,17 +1245,14 @@
    "metadata": {},
    "outputs": [
     {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "Converted 00_constructor.ipynb.\n",
-      "Converted 01_layers.ipynb.\n",
-      "Converted 02_resnet.ipynb.\n",
-      "Converted 03_xresnet.ipynb.\n",
-      "Converted 04_Net.ipynb.\n",
-      "Converted 80_test_net.ipynb.\n",
-      "Converted 81_test_xresnet.ipynb.\n",
-      "Converted index.ipynb.\n"
+     "ename": "ModuleNotFoundError",
+     "evalue": "No module named 'nbdev'",
+     "output_type": "error",
+     "traceback": [
+      "\u001b[0;31m---------------------------------------------------------------------------\u001b[0m",
+      "\u001b[0;31mModuleNotFoundError\u001b[0m                       Traceback (most recent call last)",
+      "\u001b[0;32m<ipython-input-17-ab238171e0ea>\u001b[0m in \u001b[0;36m<module>\u001b[0;34m\u001b[0m\n\u001b[1;32m      1\u001b[0m \u001b[0;31m# hide\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[0;32m----> 2\u001b[0;31m \u001b[0;32mfrom\u001b[0m \u001b[0mnbdev\u001b[0m\u001b[0;34m.\u001b[0m\u001b[0mexport\u001b[0m \u001b[0;32mimport\u001b[0m \u001b[0;34m*\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[0m\u001b[1;32m      3\u001b[0m \u001b[0mnotebook2script\u001b[0m\u001b[0;34m(\u001b[0m\u001b[0;34m)\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n",
+      "\u001b[0;31mModuleNotFoundError\u001b[0m: No module named 'nbdev'"
      ]
     }
    ],