소스 검색

typo in res net

Gen TANG 1 년 전
부모
커밋
113a06fe50
1개의 변경된 파일2개의 추가작업 그리고 2개의 파일을 삭제
  1. 2 2
      ch09_cnn/res_nets.ipynb

+ 2 - 2
ch09_cnn/res_nets.ipynb

@@ -252,8 +252,8 @@
     "        x = self.block1(x) # (B,  20, 28, 28)\n",
     "        x = self.block2(x) # (B,  40, 14, 14)\n",
     "        x = self.block3(x) # (B,  60,  7,  7)\n",
-    "        x = self.block4(x) # (B,  60,  4,  4)\n",
-    "        x = self.block5(x) # (B,  60,  2,  2)\n",
+    "        x = self.block4(x) # (B,  80,  4,  4)\n",
+    "        x = self.block5(x) # (B, 100,  2,  2)\n",
     "        x = self.block6(x) # (B, 120,  1,  1)\n",
     "        out = self.lm(x.view(x.shape[0], -1)) # (B, 10)\n",
     "        return out\n",