|
|
@@ -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",
|