Browse Source

typo in res net

Gen TANG 1 year ago
parent
commit
113a06fe50
1 changed files with 2 additions and 2 deletions
  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",