fix pad size in Conv2d layer (yandexdataschool#25)

simflin · web-flow · commit 3dea61e62ff6 · 2019-02-19T16:17:04.000+03:00
diff --git a/homework01/homework_modules.ipynb b/homework01/homework_modules.ipynb
@@ -1018,7 +1018,7 @@
     "        self.gradb = np.zeros_like(self.b)\n",
     "        \n",
     "    def updateOutput(self, input):\n",
-    "        pad_size = self.kernel_size % 2\n",
+    "        pad_size = self.kernel_size // 2\n",
     "        # YOUR CODE ##############################\n",
     "        # 1. zero-pad the input array\n",
     "        # 2. compute convolution using scipy.signal.correlate(... , mode='valid')\n",
@@ -1029,7 +1029,7 @@
     "        return self.output\n",
     "    \n",
     "    def updateGradInput(self, input, gradOutput):\n",
-    "        pad_size = self.kernel_size % 2\n",
+    "        pad_size = self.kernel_size // 2\n",
     "        # YOUR CODE ##############################\n",
     "        # 1. zero-pad the gradOutput\n",
     "        # 2. compute 'self.gradInput' value using scipy.signal.correlate(... , mode='valid')\n",
@@ -1039,7 +1039,7 @@
     "        return self.gradInput\n",
     "    \n",
     "    def accGradParameters(self, input, gradOutput):\n",
-    "        pad_size = self.kernel_size % 2\n",
+    "        pad_size = self.kernel_size // 2\n",
     "        # YOUR CODE #############\n",
     "        # 1. zero-pad the input\n",
     "        # 2. compute 'self.gradW' using scipy.signal.correlate(... , mode='valid')\n",
