From fc59e3d3df3192c2f0a4d81099c19e4f97156455 Mon Sep 17 00:00:00 2001 From: cy Date: Sun, 18 Mar 2018 21:03:52 +0800 Subject: [PATCH] fix error in resnet34 of chapter4 --- .../chapter4.ipynb" | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git "a/chapter4-\347\245\236\347\273\217\347\275\221\347\273\234\345\267\245\345\205\267\347\256\261nn/chapter4.ipynb" "b/chapter4-\347\245\236\347\273\217\347\275\221\347\273\234\345\267\245\345\205\267\347\256\261nn/chapter4.ipynb" index d90310b4..c7d36031 100644 --- "a/chapter4-\347\245\236\347\273\217\347\275\221\347\273\234\345\267\245\345\205\267\347\256\261nn/chapter4.ipynb" +++ "b/chapter4-\347\245\236\347\273\217\347\275\221\347\273\234\345\267\245\345\205\267\347\256\261nn/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",