fix error in resnet34 of chapter4

chenyuntc · Mar 18, 2018 · fc59e3d · fc59e3d
1 parent 584251c
commit fc59e3d
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/chapter4-神经网络工具箱nn/chapter4.ipynb b/chapter4-神经网络工具箱nn/chapter4.ipynb
@@ -1871,10 +1871,10 @@
     "                nn.MaxPool2d(3, 2, 1))\n",
     "        \n",
     "        # 重复的layer，分别有3，4，6，3个residual block\n",
-    "        self.layer1 = self._make_layer( 64, 128, 3)\n",
-    "        self.layer2 = self._make_layer( 128, 256, 4, stride=2)\n",
-    "        self.layer3 = self._make_layer( 256, 512, 6, stride=2)\n",
-    "        self.layer4 = self._make_layer( 512, 512, 3, stride=2)\n",
+    "        self.layer1 = self._make_layer( 64, 64, 3)\n",
+    "        self.layer2 = self._make_layer( 64, 128, 4, stride=2)\n",
+    "        self.layer3 = self._make_layer( 128, 256, 6, stride=2)\n",
+    "        self.layer4 = self._make_layer( 256, 512, 3, stride=2)\n",
     "\n",
     "        #分类用的全连接\n",
     "        self.fc = nn.Linear(512, num_classes)\n",