Merge pull request #84 from ayasyrev/ayasyrev/issue78

Ayasyrev/issue78

Author: ayasyrev

Nbs/00_ModelConstructor.ipynb

@@ -334,7 +334,7 @@
        "ConvBnAct(\n",
        "  (conv): Conv2d(32, 64, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1), bias=False)\n",
        "  (bn): BatchNorm2d(64, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
-       "  (act_fn): LeakyReLU(negative_slope=0.01)\n",
+       "  (act_fn): LeakyReLU(negative_slope=0.01, inplace=True)\n",
        ")"
       ]
      },
@@ -345,7 +345,7 @@
    ],
    "source": [
     "#collapse_output\n",
-    "conv_layer = ConvBnAct(32, 64, bn_1st=True, act_fn=nn.LeakyReLU())\n",
+    "conv_layer = ConvBnAct(32, 64, bn_1st=True, act_fn=nn.LeakyReLU)\n",
     "conv_layer"
    ]
   },
@@ -423,6 +423,13 @@
     "# print_doc(SEModule)"
    ]
   },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": []
+  },
   {
    "cell_type": "code",
    "execution_count": null,